import pandas as pd

from MySQLConfig import SO # SOTorrent18_12 configure .py file


### MySQLのクエリを引数に与えるとデータフレームで返ってくるようにしています ###

class SQL:
    
    # redirect query
    def so_gets(query): # queryにクエリの文字列を入れる
        try:
            print(f'   Query: {query}')
            print('Connecting...')
            data = pd.read_sql(query, SO.config())
            print('Successfuly!!!')
            print(f'Return Data Rows：{len(data)}×{len(data.columns)}')
            return data
        
        except Exception as e:
            print(e)
    
    
    # redirect query over ssh
    def ssh_so_gets(query): # queryにクエリの文字列を入れる
        try:
            print(f'   Query: {query}')
            print('Connecting...')
            data = pd.read_sql(query, SO.ssh_config())
            print('Successfuly!!!')
            print(f'Return Data Rows：{len(data)}×{len(data.columns)}')
            return data
        
        except Exception as e:
            print(e)

            
def main():


    ###############################################################
    #                                                             #
    #        　　　　処理内容をここに記述してください                    #
    #                                                             #
    ###############################################################


    
    #サンプルコード
    print('Query:')
    query = input()
    print(query)
    df = SQL.so_gets(query) # 引数にクエリを与える必要があります
    print(df)
    print()
        

if __name__ == '__main__':
    main()