Moving mta_manager logic into its own package - will most likely mvoe to its own repo shortly
This commit is contained in:
59
mta_manager/mta_test.py
Normal file
59
mta_manager/mta_test.py
Normal file
@@ -0,0 +1,59 @@
|
||||
import os
|
||||
from dotenv import load_dotenv
|
||||
from mta import MTA
|
||||
import threading
|
||||
import time
|
||||
from time import sleep
|
||||
from pprint import pprint
|
||||
|
||||
load_dotenv()
|
||||
|
||||
api_key = os.getenv('MTA_API_KEY', '')
|
||||
mtaController = MTA(
|
||||
api_key,
|
||||
["A", "C", "E", "1", "2", "3"],
|
||||
["127S", "127N", "A27N", "A27S"]
|
||||
)
|
||||
|
||||
|
||||
|
||||
async def mta_callback(routes):
|
||||
print("We are inside of the call back now")
|
||||
print(len(routes))
|
||||
pprint(mtaController.convert_routes_to_station_first(routes))
|
||||
|
||||
class threadWrapper(threading.Thread):
|
||||
def __init__(self, run):
|
||||
threading.Thread.__init__(self)
|
||||
self.run = run
|
||||
|
||||
def run(self):
|
||||
self.run()
|
||||
|
||||
def start_mta():
|
||||
mtaController.add_callback(mta_callback)
|
||||
mtaController.start_updates()
|
||||
|
||||
def stop_mta():
|
||||
sleep(10)
|
||||
mtaController.stop_updates()
|
||||
|
||||
threadLock = threading.Lock()
|
||||
threads = []
|
||||
|
||||
# Create new threads
|
||||
thread1 = threadWrapper(start_mta)
|
||||
thread2 = threadWrapper(stop_mta)
|
||||
|
||||
|
||||
thread1.start()
|
||||
thread2.start()
|
||||
|
||||
# Add threads to thread list
|
||||
threads.append(thread1)
|
||||
threads.append(thread2)
|
||||
|
||||
# Wait for all threads to complete
|
||||
for t in threads:
|
||||
t.join()
|
||||
print ("Exiting Main Thread")
|
||||
Reference in New Issue
Block a user