presik_pos/app/localdb.py
2020-09-06 21:19:19 -05:00

144 lines
4.5 KiB
Python

import os
import sqlite3
file_db = 'app.db'
if os.name == 'posix':
home = 'HOME'
directory = '.tryton'
elif os.name == 'nt':
home = 'USERPROFILE'
directory = 'AppData/Local/tryton'
_directory = os.path.join(os.getenv(home), directory)
localfile = os.path.join(_directory, file_db)
class LocalStore(object):
def __init__(self, params=None):
self.conn = sqlite3.connect(localfile)
def create_table_config(self):
cursorObj = self.conn.cursor()
cursorObj.execute("CREATE TABLE IF NOT EXISTS configuration( \
'id' INTEGER PRIMARY KEY, \
'synchronized_date' DATETIME, \
'company_id' INTEGER, \
'company_name' TEXT, \
'defualt_party_id' INTEGER, \
'defualt_party_name' TEXT \
)")
self.conn.commit()
def create_table_product(self):
cur = self.conn.cursor()
cur.execute("CREATE TABLE IF NOT EXISTS product( \
'id' INTEGER PRIMARY KEY, \
'code' TEXT, \
'write_date' DATE, \
'barcode' TEXT, \
'template.name' TEXT, \
'description' TEXT, \
'quantity' TEXT, \
'template.account_category' INTEGER, \
'template.sale_price_w_tax' REAL \
)")
self.conn.commit()
def get_config(self):
cur = self.conn.cursor()
cur.execute("SELECT id, synchronized_date FROM configuration")
config_ = cur.fetchall()
if config_:
return config_[0]
def set_config(self, config):
cur = self.conn.cursor()
cur.execute("INSERT INTO configuration(id, synchronized_date, company_id) \
VALUES(1, '2010-01-01 00:00:00', ?)", config)
self.conn.commit()
def set_config_sync(self, config_dt):
cur = self.conn.cursor()
cur.execute("UPDATE configuration SET synchronized_date=?", [config_dt])
self.conn.commit()
def get_local_products(self):
cur = self.conn.cursor()
cur.execute("SELECT id FROM product")
rows = cur.fetchall()
rows_ = [r[0] for r in rows]
return rows_
def update_products(self, products, local_products):
len_products = len(products)
print('------- Creating %d products -------' % len_products)
to_create = []
to_update = []
for p in products:
if p[0] not in local_products:
to_create.append(p)
else:
to_update.append([p[1], p[4], float(p[7]), p[6], p[0]])
if to_create:
cur = self.conn.cursor()
cur.executemany("INSERT INTO product(id, write_date, code, \
barcode, 'template.name', description, \
'template.account_category', 'template.sale_price_w_tax') \
VALUES(?, ?, ?, ?, ?, ?, ?, ?)", to_create)
self.conn.commit()
if to_update:
cur = self.conn.cursor()
for rec in to_update:
cur.execute('UPDATE product SET write_date=?, "template.name"=?,\
"template.sale_price_w_tax"=?, "template.account_category"=? \
WHERE id=?', rec)
self.conn.commit()
def find_product_by_id(self, product_id):
cur = self.conn.cursor()
query = "SELECT * FROM product WHERE id=?"
cur.execute(query, (product_id,))
res = cur.fetchall()
return res[0]
def find_product_elastic(self, domain, limit):
self.conn.row_factory = lambda c, r: dict(
zip([col[0] for col in c.description], r)
)
cur = self.conn.cursor()
clauses = []
for d in domain:
clause1 = '"{}" LIKE "{}"'.format(d[1][0], d[1][2])
clause2 = '"{}" LIKE "{}"'.format(d[2][0], d[2][2])
clause3 = '(' + clause1 + ' OR ' + clause2 + ')'
clauses.append(clause3.upper())
if len(clauses) > 1:
_clause = ' AND '.join(clauses)
else:
_clause = clauses[0]
query = "SELECT * FROM product WHERE {}".format(_clause)
cur.execute(query)
rows = cur.fetchall()
return rows
# def do_query(self):
# c = self.conn.cursor()
# c.execute(query)
# rows = c.fetchall()
# self.conn.commit()
# self.conn.close()
# for row in rows:
# print(row)
if __name__ == "__main__":
store = LocalStore()
store.create_table()
store.get_local_products()