def close_connection_to_database(cursor: psycopg2.extensions.cursor, connection: psycopg2.extensions.connection): try: # closing database connection. if connection: cursor.close() connection.close() print("PostgreSQL connection is closed") except (Exception, psycopg2.Error) as error: print(f"Error while disconnection from PostgreSQL: {error}")
def create_table(table_name:str,column_detail_nameList:list,conn:psycopg2.extensions.connection): try: cursor=conn.cursor() create_table_query= "CREATE TABLE "+table_name+" ("; cnt=0 for col_detail in column_detail_nameList: if cnt==0: create_table_query+= " "+ col_detail cnt=1 else: create_table_query+= " ,"+ col_detail create_table_query+=" );" cursor.execute(create_table_query) conn.commit() print(table_name,"-Table created successfully in PostgreSQL ") except (Exception, psycopg2.DatabaseError) as error : print ("Error while creating PostgreSQL table", error) finally: #closing database connection. if(conn): cursor.close() conn.close() print("PostgreSQL connection is closed")