From 0ee00536e6c1a64dadc4da17a545d3c17165fa64 Mon Sep 17 00:00:00 2001 From: Victor Häggqvist Date: Mon, 18 May 2015 19:12:50 +0200 Subject: update to support python 3 --- xboomx/sqlitemgr.py | 38 ++++++++++++++++++++++++++++++++++++++ 1 file changed, 38 insertions(+) create mode 100644 xboomx/sqlitemgr.py (limited to 'xboomx/sqlitemgr.py') diff --git a/xboomx/sqlitemgr.py b/xboomx/sqlitemgr.py new file mode 100644 index 0000000..6e05bb2 --- /dev/null +++ b/xboomx/sqlitemgr.py @@ -0,0 +1,38 @@ +# coding=utf-8 +import os +from sqlalchemy import create_engine +from sqlalchemy.ext.declarative import declarative_base +from sqlalchemy.orm import sessionmaker +from sqlalchemy import Column, Integer, String + +__author__ = 'Victor Häggqvist' + +# create dir if not exists +try: + os.makedirs(os.getenv("HOME") + '/.xboomx') +except: + pass + +dbname = 'xboomx_sqlite.db' +dbpath = os.path.join(os.getenv("HOME"), '.xboomx', dbname) +dsn = 'sqlite:///%s' % dbpath + +engine = create_engine(dsn, echo=False) +Base = declarative_base() + + +class PathItem(Base): + __tablename__ = 'pathitems' + id = Column(Integer, primary_key=True) + name = Column(String) + count = Column(Integer) + + def __repr__(self): + return "" % (self.name, self.count) + + +def get_session(): + Base.metadata.create_all(engine) + Session = sessionmaker(bind=engine) + session = Session() + return session -- cgit v1.2.3