No issue: add main function to setup-startup-profiling.

tools/setup-startup-profiling.py

@@ -48,17 +48,21 @@ def remove(filename):
     run(['adb', 'shell', 'rm', PATH_PREFIX + '/' + filename])
     run(['adb', 'shell', 'am', 'clear-debug-app'])
 
-try:
+def main():
-    cmd = sys.argv[1]
+    try:
-    id = sys.argv[2]
+        cmd = sys.argv[1]
-except IndexError as e:
+        id = sys.argv[2]
-    print_usage_and_exit()
+    except IndexError as e:
+        print_usage_and_exit()
+
+    filename = id + '-geckoview-config.yaml'
 
-filename = id + '-geckoview-config.yaml'
+    if cmd == 'push':
+        push(id, filename)
+    elif cmd == 'remove':
+        remove(filename)
+    else:
+        print_usage_and_exit()
 
-if cmd == 'push':
+if __name__ == '__main__':
-    push(id, filename)
+    main()
-elif cmd == 'remove':
-    remove(filename)
-else:
-    print_usage_and_exit()