def __init__(self):

# ADJUST THIS TO ALLOW FOR SLOWER OR FASTER MOVEMENTS

# DELAY MAY NEED TO BE INCREASED BASED ON INTERNET SPEED AND AGGREGATION/LENGTH OF CHARTS

# THIS IS NEEDED BECAUSE SOMETIMES THE CHARTS DO NOT LOAD FAST ENOUGH

self.delay = 0.15 # 0.15

# THIS IS THE WATCHLIST FILE FROM TOS, WHICH IS FORMATED WITH THE CURRENT DATE OF WHICH DOWNLOADED

self.watchlist_file = f"{datetime.now().strftime('%Y-%m-%d')}-watchlist.csv"

# PATH TO USERS DOCUMENTS FOLDER. THIS IS WHERE TOS DOWNLOADS THE STRATEGY REPORTS

self.path = "C:/Users/TreyT/Documents"

# TOTAL NUMBER OF SYMBOLS IN WATCHLIST

self.watchlist_length = 0

self.watchlist_length_copy = 0

self.watchlist_symbols = []

# ADDED CONDITIONAL FOR THE WHILE LOOP IN START METHOD.

# IF ERROR, ATTRIBUTE WILL BE SET TO FALSE AND WHILE LOOP WILL STOP

self.no_error = True

self.on_start = True

def throwError(self, error):

self.no_error = False

for _ in range(3):

playsound('audio/error.mp3')

print(error)

# REMOVE ALL EXISTING STRATEGY REPORTS CSV FILES FROM DOCUMENTS FOLDER

for file in os.listdir(export.path):

sep = file.split("_")

if file.endswith(".csv") and sep[0].strip() == "StrategyReports":

try:

os.remove(os.path.join(export.path, file))

except:

pass

def getWatchlist(self):

# MOVES TO WATCHLIST HAMBURGER ICON AND CLICKS

pyautogui.click(282, 144) # ABSOLUTE

time.sleep(self.delay)

# MOVES DOWN DROPDOWN LIST TO Export to file... TAB

pyautogui.move(0, 155) # RELATIVE

pyautogui.click()

time.sleep(self.delay)

# MOVES OVER TO THE Save BUTTON ON THE FILE EXPLORER TO SAVE WATCHLIST.

# IN THE FILE EXPLORER BEFORE ANYTHING ELSE, CHANGE YOUR DIRECTORY TO YOUR WORKING DIRECTORY. THIS WILL BE THE NEW DEFAULT DIRECTORY FOR NOW ON.

pyautogui.move(500, 215) # RELATIVE

time.sleep(self.delay)

pyautogui.click()

time.sleep(self.delay)

if os.path.exists(self.watchlist_file):

print("FOUND WATCHLIST!")

with open(self.watchlist_file, "r") as watchlist:

reader = csv.reader(watchlist)

# THESE 4 NEXTS START THE ITERATION AT THE NEEDED ROWS WITH THE SYMBOL DATA

# MAY NEED TO BE ADJUSTED BASED ON USERS NEEDS

next(reader)

next(reader)

next(reader)

next(reader)

for row in reader:

self.watchlist_length += 1

self.watchlist_length_copy += 1

self.watchlist_symbols.append(row[0])

os.remove(self.watchlist_file)

print("SYMBOLS EXTRACTED FROM WATCHLIST!\n")

# MOVES CURSOR TO WATCHLIST AND CLICKS ON TOP MOST SYMBOL IN LIST

pyautogui.click(150, 174) # ABSOLUTE

# CHECK WHERE WE ARE IN ITERATION, AND ARROW DOWN IF NEED BE

for _ in range(len(

[file for file in os.listdir("csv_files")])):

pyautogui.press(['down'])

# SUBTRACT ALREADY EXISTING COUNT IN CSV FILES DIRECTORY FROM WATCHLIST LENGTH

self.watchlist_length -= len(

[file for file in os.listdir("csv_files")])

proceed = input("Proceed? (y/n): ")

if proceed.upper() == "Y":

print("BEGINNING AUTO EXPORTS....")

self.start()

def moveCSVFile(self):

# ABSOLUTE PATH TO CSV_FILES DIRECTORY

dst_dir = "C:/Users/TreyT/Desktop/AlgoTrading/TOS_Auto_Export/csv_files"

for root, dirs, files in os.walk(self.path):

for f in files:

sep = f.split("_")

if f.endswith('.csv') and sep[0].strip() == "StrategyReports":

try:

# MOVES MOST RECENT STRATEGY REPORT FILE FROM DOCUMENTS FOLDER TO CSV_FILES FOLDER

shutil.move(os.path.join(root, f), dst_dir)

# SUBSTRACTS ONE FROM WATCHLIST_LENGTH ATTRIBUTE

# THIS TELLS US HOW MANY MORE FILES TO EXPECT

self.watchlist_length -= 1

# RESET WATCHLIST_LENGTH IF PROGRAM RESTART FROM DIFFERENT SYMBOL

if self.on_start:

symbol = sep[1].strip()

self.watchlist_length = self.watchlist_length_copy - (self.watchlist_symbols.index(symbol) + 1)

self.on_start = False

except shutil.Error as e:

self.throwError(e)

return

# IF MISS, THROW ERROR

self.throwError("MISSED EXPORT - STOPPING PROGRAM - PLEASE RESTART")

def start(self):

start = time.perf_counter()

# GET THE SIZE OF THE MONITOR THAT WILL DISPLAY THE TOS APP

screen_width, screen_length = pyautogui.size()

# START_X AND START_Y ARE BOTH WHERE THE MOUSE ARROW WILL START ON SCREEN.

# THIS SHOULD BE PLACED WITHIN THE SMA BAR GRAPH TO BE ABLE TO GET THE REPORTS.

# WILL NEED TO BE ADJUSTED BASED ON SCREEN SIZE

start_x = int(screen_width / 2.660194174757281)

start_y = int(screen_length / 1.728)

# GET AVERAGE ITERATION CYCLE TIME AND GET ESTIMATED TIME REMAINING

cycles = []

previous_cycle_time = None

while self.watchlist_length != 0 and self.no_error:

# SHOW REPORT BUTTON

pyautogui.rightClick(start_x, start_y) # ABSOLUTE

time.sleep(self.delay)

pyautogui.move(30, 90) # RELATIVE

time.sleep(self.delay + 0.25)

pyautogui.click()

# EXPORT FILE BUTTON

pyautogui.move(520, 40) # RELATIVE

time.sleep(self.delay)

pyautogui.click()

# SAVE BUTTON (save strategy report to documents folder)

pyautogui.move(-160, -90) # RELATIVE

time.sleep(self.delay)

pyautogui.click()

# # overwrite

# pyautogui.move(-85, -575) # RELATIVE

# time.sleep(self.delay)

# pyautogui.click()

time.sleep(self.delay)

# CLOSE BUTTON (close out of the strategy report)

pyautogui.move(200, 100) # RELATIVE

time.sleep(self.delay)

pyautogui.click()

time.sleep(self.delay)

# NEXT SYMBOL (arrows down to next symbol in watchlist)

pyautogui.press(['down'])

time.sleep(self.delay + .3)

###############

self.moveCSVFile()

if self.no_error:

middle = time.perf_counter()

running_time = time.strftime(

'%H:%M:%S', time.gmtime(middle - start))

if len(cycles) > 2:

time_remaining = time.strftime(

'%H:%M:%S', time.gmtime(statistics.mean(cycles) * self.watchlist_length))

else:

time_remaining = "N/A"

print(

f"REMAINING --> {self.watchlist_length} | RUNNING TIME: {running_time} | EST. TIME REMAINING: {time_remaining}")

if previous_cycle_time != None:

cycles.append((middle - start) - previous_cycle_time)

previous_cycle_time = middle - start

if self.no_error:

end = time.perf_counter()

# FINISHED

print("\nFINISHED!")

playsound("audio/alert_on_call.mp3")

took = time.strftime('%H:%M:%S', time.gmtime(end - start))