Advertisement
Guest User

Untitled

a guest
Mar 31st, 2020
139
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 1.84 KB | None | 0 0
  1. #!/usr/bin/python
  2. import sys
  3.  
  4. import getopt
  5. from datetime import datetime
  6.  
  7. import pandas as pd
  8.  
  9. from sqlalchemy import create_engine
  10.  
  11. if __name__ == "__main__":
  12.  
  13. # укажите входные параметры в строках
  14. unixOptions = "std:edt"
  15. gnuOptions = ["start_dt=", "end_dt"]
  16.  
  17. fullCmdArguments = sys.argv
  18. argumentList = fullCmdArguments[1:]
  19.  
  20. try:
  21. arguments, values = getopt.getopt(argumentList, unixOptions, gnuOptions)
  22. except getopt.error as err:
  23. print (str(err))
  24. sys.exit(2)
  25.  
  26. start_dt = '1981-01-01'
  27. end_dt = '1998-01-01'
  28. for currentArgument, currentValue in arguments:
  29. if currentArgument in ("-sdt", "--start_dt"):
  30. start_dt = currentValue
  31. elif currentArgument in ("-edt", "--end_dt"):
  32. end_dt = currentValue
  33.  
  34. db_config = {'user': 'my_user',
  35. 'pwd': 'my_user_password',
  36. 'host': 'localhost',
  37. 'port': 5432,
  38. 'db': 'games'}
  39. connection_string = 'postgresql://{}:{}@{}:{}/{}'.format(db_config['user'],
  40. db_config['pwd'],
  41. db_config['host'],
  42. db_config['port'],
  43. db_config['db'])
  44.  
  45. engine = create_engine(connection_string)
  46.  
  47. query = ''' SELECT *
  48. FROM data_raw
  49. WHERE year_of_release::TIMESTAMP BETWEEN '{}'::TIMESTAMP AND '{}'::TIMESTAMP
  50. '''.format(start_dt, end_dt)
  51.  
  52. data_raw = pd.io.sql.read_sql(query, con = engine, index_col = 'game_id')
  53. print(data_raw['year_of_release'].unique())
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement