glasslion · April 15, 2019 13:00 · Apr 15, 2019
diff --git a/bilibili_screenshots.py b/bilibili_screenshots.py
@@ -0,0 +1,72 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+"""
+Bilibili Screenshots Syncer
+
+It can download `/Pictures/bili/screenshot` from a Android phone via ftp.
+It also rename the picture so the timestamp comes first.
+"""
+from __future__ import print_function
+import argparse
+from ftplib import FTP
+import glob
+import re
+import os
+import subprocess
+
+try:
+    input = raw_input
+except NameError:
+    pass
+
+
+# -r: Is for recursively download.
+# -np: Is for no parent ascending.
+# -nH: Is for disabling creation of directory having name same as URL i.e. abc.xyz.com
+# --cut-dirs: Is for ignoring number of parent directories. e.g. Pictures/bili/screenshot/*.png has 3 parent dirs
+# -N: Is for skiping existing files
+WGET_CMD = "wget  -r -N -np -nH --cut-dirs=2 ftp://{ip}:{port}/Pictures/bili/screenshot"
+
+
+def download_pics(ip, port):
+    cmd = WGET_CMD.format(ip=ip, port=port)
+    subprocess.call(cmd, shell=True)
+
+
+def rename_file():
+    pic_paths = glob.glob('screenshot/*.png')
+    for path in pic_paths:
+        new_path = re.sub(r"(\d+)@(\d+)@(.*.png)", r"\g<2>_\g<1>_\g<3>", path)
+        os.rename(path, new_path)
+
+def yes_no(question_to_be_answered):
+    while True:
+        choice = input(question_to_be_answered  + ' [y/n]: ').lower()
+        if choice[:1] == 'y':
+            return True
+        elif choice[:1] == 'n':
+            return False
+        else:
+            print("Please respond with 'Yes' or 'No'\n")
+
+def delete_remote(ip, port):
+    ftp = FTP()
+    ftp.connect(ip, port)
+    ftp.login()
+    if yes_no('Delete the remote dir /Pictures/bili/screenshot?'):
+        ftp.rmd('/Pictures/bili/screenshot')
+
+
+def main():
+    parser = argparse.ArgumentParser(description='Bilibili Screenshots Syncer')
+    parser.add_argument('ip', help='FTP server IP')
+    parser.add_argument('--port', default=2121, type=int, help='FTP server port')
+    args = parser.parse_args()
+
+    download_pics(args.ip, args.port)
+    rename_file()
+    delete_remote(args.ip, args.port)
+
+
+if __name__ == '__main__':
+    main()