コード例 #1
0
ファイル: connection.py プロジェクト: gferrazoli/Coconut-Bot
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}")
コード例 #2
0
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")