Guest User

Untitled

a guest
Sep 26th, 2018
104
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 0.90 KB | None | 0 0
  1. import csv
  2. import xlrd
  3. import os
  4. import psycopg2
  5. import django
  6. from simplemostmeals import settings
  7. django.setup()
  8. from core import models
  9.  
  10.  
  11. try:
  12. conn = psycopg2.connect("host=localhost dbname=prodmealsdb user=postgres password=blank")
  13. cur = conn.cursor()
  14.  
  15. filepath = '/path/to/your/data_to_be_updated.csv'
  16. ext = os.path.splitext(filepath)[-1].lower()
  17. if (ext == '.csv'):
  18. print("csv file")
  19. with open(filepath) as csvfile:
  20. next(csvfile)
  21. readCSV = csv.reader(csvfile, delimiter=',')
  22. for row in readCSV:
  23. print(row[3],row[5])
  24. cur.execute("UPDATE your_table SET column_to_be_updated = %s where id = %s", (row[5], row[3]))
  25. conn.commit()
  26. conn.close()
  27. cur.close()
  28.  
  29. except (Exception, psycopg2.DatabaseError) as error:
  30. print(error)
  31. finally:
  32. if conn is not None:
  33. conn.close()
Add Comment
Please, Sign In to add comment