Add top 300 power mode

Add a mode to grab only power from the top 300 players. Run with --top
to use this mode, or --power for full power rankings. Flag names should
be changed.

.gitignore

@@ -0,0 +1,2 @@
+output/
+debug/

rok-stat-grabber.py

@@ -30,11 +30,14 @@ PROGRAM_DESCRIPTION = "This program controls devices via ADB to take screenshots
 # ----
 
 parser = argparse.ArgumentParser(description = PROGRAM_DESCRIPTION)
+parser.add_argument("--power", help = "get full power rankings", default = False, action = "store_true")
+parser.add_argument("--top", help = "get stats on the top 300", default = False, action = "store_true")
 parser.add_argument("-p", "--project", help = "project name", required = True)
 parser.add_argument("-s", "--serial", help = "device serial", required = True)
 parser.add_argument("-v", "--verbose", help = "be verbose", default = False, action = "store_true")
-parser.add_argument("-c", "--count", help = "how many to scrape", required = True, type = int)
+parser.add_argument("-c", "--count", help = "how many to scrape", default = 0, type = int)
 parser.add_argument("-n", "--start", help = "what number to start scraping at", default = 1, type = int)
+parser.add_argument("-k", "--skipped", help = "how many have been skipped", type = int)
 parser.add_argument("--debug", help = "debug", default = False, action = "store_true")
 arguments = parser.parse_args()
 
@@ -42,7 +45,7 @@ arguments = parser.parse_args()
 # Coordinates
 # ----
 
-ROTATE = 0 # TODO: fix this; shouldn't be needed
+ROTATE = 90 # TODO: fix this; shouldn't be needed
 
 TAP_LOCATIONS = {
     "1": (460, 460),
@@ -178,6 +181,51 @@ def read_from_power_rankings(count, start, device, folder):
         if i%25 == 0: time.sleep(10)
     return
 
+def read_top_300_stats(start, skipped, device, folder):
+    if arguments.verbose: print("Reading the top 300 power rankings.")
+    if skipped > 0: print("Adding", skipped, "already skipped rankings.")
+    time.sleep(1)
+
+    count = 300 + 5 + skipped
+    i = start
+    while i < (count + 1):
+        if arguments.verbose: print(str(i) + "/" + str(count) + ": ", end = "", flush = True)
+
+        # Scrape
+        tap_modifier = tap_profile(i, folder)
+        i = i + tap_modifier
+        count = count + tap_modifier - 1
+        grab_profile_simple(device, folder)
+
+        # Rest
+        time.sleep(1)
+        if i%10 == 0: time.sleep(5)
+        if i%25 == 0: time.sleep(10)
+    return
+
+def grab_profile_simple(device, folder):
+    datetime = get_datetime_string()
+
+    # Screenshot profile
+    # ID, Username, Power, Kill Points
+    time.sleep(2)
+    take_screenshot(device, folder + datetime + "_profile")
+
+    # Copy and save username
+    device.shell("input tap 1055 455")
+    time.sleep(1)
+    username = get_clipboard(device)
+    with open(projectFolder + "usernames.txt", 'a+', newline='', encoding='utf-8') as output_file:
+        output_file.write(datetime + "\t")
+        output_file.write(username)
+        output_file.write("\n")
+    if arguments.verbose: print(username, end = "", flush = True)
+
+    # Exit
+    if arguments.verbose: print("") # TODO: say "done" or something here if successful
+    device.shell("input tap 2185 160")
+    return
+
 # ----
 # Main
 # ----
@@ -197,4 +245,5 @@ if __name__ == '__main__':
     if arguments.debug: take_screenshot(device, projectFolder + get_datetime_string() + "_debug")
 
     # Scrape
-    read_from_power_rankings(arguments.count, arguments.start, device, projectFolder)
+    if arguments.power: read_from_power_rankings(arguments.count, arguments.start, device, projectFolder)
+    if arguments.top: read_top_300_stats(arguments.start, arguments.skipped, device, projectFolder)