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