Advertisement
Guest User

Untitled

a guest
Dec 26th, 2016
82
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 1.15 KB | None | 0 0
  1. import psycopg2
  2. import random
  3. import string
  4.  
  5. def random_word():
  6. return ''.join(random.choice(string.ascii_uppercase + string.ascii_lowercase) for _ in range(random.randint(5, 15)))
  7.  
  8. def rand_gender():
  9. gender = random.randint(0,1)
  10. genders = { '0' : "male", '1' : "female"}
  11. return genders[str(gender)]
  12.  
  13. def main():
  14. conn_string = "host='localhost' dbname='postgres' user='postgres' password='postgres'"
  15.  
  16. print("Connecting to database\n ->%s" % (conn_string))
  17.  
  18. conn = psycopg2.connect(conn_string)
  19.  
  20. cursor = conn.cursor()
  21. cursor.execute("CREATE TABLE IF NOT EXISTS users (id INT NOT NULL, first_name VARCHAR, last_name VARCHAR, email VARCHAR, gender VARCHAR)")
  22. print("Connected!\n")
  23.  
  24. for i in range(1000000):
  25. qs = "INSERT INTO users (id, first_name, last_name, email, gender) VALUES "
  26. query_string = "('{0}', '{1}', '{2}', '{3}@{4}.com', '{5}')"
  27. query = query_string.format(i, random_word(), random_word(), random_word(), random_word(), rand_gender())
  28. qs = qs + query
  29. if i%1000 == 0: print (str(i) + "row is added")
  30. res = cursor.execute(qs)
  31. conn.commit()
  32.  
  33. cursor.close()
  34. conn.close()
  35.  
  36. if __name__ == "__main__":
  37. main()
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement