def _load_deps(self, document_id):
conn = None
try:
conn = self.pool.getconn()
pgcast.register(conn)
curs = conn.cursor()
curs.execute(self.GET_DEPS_SQL, (document_id, ))
self.deps = curs.fetchone()[0]
conn.commit()
curs.close()
except (Exception, psycopg2.DatabaseError) as error:
print(error)
finally:
if conn is not None:
self.pool.putconn(conn)
def delete(self, document_id):
success = True
conn = None
try:
conn = self.pool.getconn()
pgcast.register(conn)
curs = conn.cursor()
curs.execute(self.DELETE_DOCUMENT_SQL, (document_id, ))
conn.commit()
curs.close()
except (Exception, psycopg2.DatabaseError) as error:
success = False
self.errors.append(error.pgerror)
finally:
if conn is not None:
self.pool.putconn(conn)
return success
def init(self):
conn = None
document_id = None
try:
conn = self.pool.getconn()
pgcast.register(conn)
curs = conn.cursor()
curs.execute(self.CREATE_DOCUMENT_SQL, (self.head, ))
document_id = curs.fetchone()[0]
conn.commit()
curs.close()
except (Exception, psycopg2.DatabaseError) as error:
self.errors.append(error.pgerror)
finally:
if conn is not None:
self.pool.putconn(conn)
return document_id
def load(self, good_code=None):
conn = None
try:
conn = self.pool.getconn()
pgcast.register(conn)
curs = conn.cursor()
if good_code:
curs.execute(self.GET_BALANCE_SQL0)
else:
curs.execute(self.GET_BALANCE_SQL0, (good_code, ))
self.body = curs.fetchall()
conn.commit()
curs.close()
except (Exception, psycopg2.DatabaseError) as error:
print(error)
finally:
if conn is not None:
self.pool.putconn(conn)
def get_document_list(self):
conn = None
document_list = None
try:
conn = self.pool.getconn()
pgcast.register(conn)
curs = conn.cursor()
curs.execute(self.GET_LSIT_SQL, (
self.facility_code,
self.date_start,
self.date_end,
))
document_list = curs.fetchone()[0]
conn.commit()
curs.close()
except (Exception, psycopg2.DatabaseError) as error:
print(error)
finally:
if conn is not None:
self.pool.putconn(conn)
return document_list