Skip to content

Instantly share code, notes, and snippets.

@kshwetabh

kshwetabh/LoadDataIntoPostgreSQL.py

Created January 11, 2018 04:06
Show Gist options

  • Select an option

    Learn more about clone URLs
  • Save kshwetabh/df21b180aafc224b8748a59e983fe578 to your computer and use it in GitHub Desktop.

Select an option

Learn more about clone URLs
Save kshwetabh/df21b180aafc224b8748a59e983fe578 to your computer and use it in GitHub Desktop.
Download ZIP

Revisions

  1. kshwetabh created this gist Jan 11, 2018.
    24 changes: 24 additions & 0 deletions LoadDataIntoPostgreSQL.py
    View file
    Original file line number Diff line number Diff line change
    @@ -0,0 +1,24 @@
    import psycopg2
    conn = psycopg2.connect("host='localhost' port='5432' dbname='stocks' user='xxxxxx' password='xxxxxx'")
    cur = conn.cursor()

    SQL_STATEMENT = """
    COPY %s FROM STDIN WITH
    CSV
    HEADER
    DELIMITER AS ','
    """

    def process_file(conn, table_name, file_object):
    cursor = conn.cursor()
    cursor.copy_expert(sql=SQL_STATEMENT % table_name, file=file_object)
    conn.commit()
    cursor.close()

    f = open(r'C:\Users\xxx\Desktop\AllStocksExtracted\code\result.csv', 'r')


    try:
    process_file(conn, 'stocks', f)
    finally:
    conn.close()