Advertisement
Guest User

Untitled

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