test.py

import sys
print(sys.path)
sys.path.append('/home/ivan/Documents/github/crobat/crobat')
import crobat 
import crobat.recorder 
import asyncio, time
#from test import test_connection

#test_connection.main()


#from datetime import datetime
import copra.rest
from copra.websocket import Channel, Client


#import pandas as pd

# # how do i want this shit to work

# # 1. you import the program
# # 2. you intialize the args, with the option to start recording
# # 3a. if you choose to start recording
# #     i. you can call the class.get_item to see what the last order was
# #     ii. you can call a function to get the current picture of the orderbook
# #     iii.  you can call a function to  get the running order book images  as dirty or pandas'd
# #     iv.  you can call a function to get the last_orders disrt or as pandas
# # 3b. if not you have the option of calling the class indiviudually within your own asyncio loop
# # 4. error handling ? 

class input_args(object):
    def __init__(self, currency_pair='ETH-USD',
                       position_range=4,
                       recording_duration=5,
                       sides=['bid','ask'],
                       filetype=['xlsx']):
    
        self.currency_pair = currency_pair
        self.position_range = position_range
        self.recording_duration = recording_duration
        self.sides = sides
        self.filetype = filetype

# def check_input(input_str, reference_var):

def main():
    settings = input_args()
    loop = asyncio.get_event_loop()
    channel = Channel('level2', settings.currency_pair) 
    channel2 =Channel('ticker', settings.currency_pair)
    ws = crobat.recorder.L2_Update(loop, channel, settings)
    ws.subscribe(channel2)
    try:
        loop.run_forever()
    except KeyboardInterrupt:
        loop.run_until_complete(ws.close())
        loop.close()

if __name__ == '__main__':
    main()