[-] misc : assertion

mos9527 · Aug 24, 2021 · 8ccde42 · 8ccde42
1 parent a9d324a
commit 8ccde42
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git a/demos/clouddrive/batchupload.py b/demos/clouddrive/batchupload.py
@@ -4,9 +4,10 @@
 from phone import login
 from time import sleep
 
-tracks = r'(your folder here)'
-if __name__ == '__main__':
-    login()
+
+if __name__ == '__main__':    
+    assert login(),"登陆失败"
+    tracks = input('[-] Path to folder:')
     for path in os.listdir(tracks):
         full_path = os.path.join(tracks,path)
         print('[-] Uploading %s' % path)

diff --git a/demos/clouddrive/cloudupload.py b/demos/clouddrive/cloudupload.py
@@ -41,7 +41,7 @@ def upload_one(path):
     print('[-] Response:\n  ',publish_result)
 
 if __name__ == "__main__":
-    login()
+    assert login(),"登陆失败"
     # login via phone,may change as you like
     upload_one(input('[-] Path to file:'))