class DBClient:
log = customlogger(globalconfig.logging_level)
def __init__(self, uri_db):
self.log.info("creating instance of DBClient: {0}".format(uri_db))
self.db = DAL(uri_db, migrate_enabled=False)
self.wells = self.db.define_table('wells',
Field('uuid'),
Field('project_uuid'),
Field('well_name'),
Field('uwi'),
Field('created_at'),
Field('modified_at'),
primarykey=['uuid'])
self.clients = self.db.define_table('clients',
Field('uuid'),
Field('company_name'),
Field('created_at'),
Field('modified_at'),
primarykey=['uuid'])
self.projects = self.db.define_table('projects',
Field('uuid'),
Field('client_uuid'),
Field('name'),
Field('created_at'),
Field('modified_at'),
Field('basin'),
Field('shapefile'),
primarykey=['uuid'])
@add_post_func_delay(0.5)
def insert_well(self, *args):
self.log.info("inserting well into DB..")
self.wells.insert(uuid=uuid.uuid4(),
well_name=args[0],
uwi=args[1],
created_at=datetime.datetime.now(),
modified_at=datetime.datetime.now())
self.db.commit()
@add_post_func_delay(0.5)
def insert_client(self, *args):
self.log.info("inserting client into DB..")
self.clients.insert(uuid=uuid.uuid4(),
company_name=args[0],
created_at=datetime.datetime.now(),
modified_at=datetime.datetime.now())
self.db.commit()
# TODO insert project has been modified on the front end such that there is no field called project type on DB table
# def insert_project(self, *args):
# self.insert_client(args[0])
# c_uuid = self.clients(company_name=args[0])['uuid']
# # self.db.projects.insert(uuid=uuid.uuid4(), name=args[1], client_uuid=c_uuid, basin='Midland', created_at=datetime.datetime.now(), modified_at=datetime.datetime.now())
# self.projects.insert(uuid=uuid.uuid4(), name=args[1], client_uuid=c_uuid, basin='midland', created_at=datetime.datetime.now(), modified_at=datetime.datetime.now())
# self.db.commit()
@add_post_func_delay(0.5)
def delete_table(self, tablename):
self.log.info("deleteing table {0}".format(tablename))
table = self.db.get(tablename)
table.truncate(mode='CASCADE')
self.db.commit()
def execute_sql(self, sql):
self.log.info("executing sql: '{0}'".format(sql))
return self.db.executesql(sql)
def close(self):
self.log.info("closing DB instance..")
self.db.close()
