feat(main): add auto_upload_minimal_size

hldh214 · Sep 17, 2020 · 20614a9 · 20614a9
1 parent 823fb5f
commit 20614a9
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/config.sample.toml b/config.sample.toml
@@ -6,6 +6,7 @@ credentials_file = ""
     [source."668668"]
     enabled = true
     auto_upload = true
+    auto_upload_minimal_size = 1024
     source_type = "huya"
     room_id = "668668"
     title = "风行云 - {datetime} - 录像"
@@ -16,6 +17,7 @@ credentials_file = ""
     [source."deer98"]
     enabled = true
     auto_upload = false
+    auto_upload_minimal_size = 1024
     source_type = "panda"
     room_id = "10535725"
     sess_key = "018c6d30a09720dd7aba6a9113fa5227bc9c78e033eb1a2be5fec88af599e3c6"

diff --git a/recorder/app.py b/recorder/app.py
@@ -67,6 +67,10 @@ def record_thread(source_type, room_id, interval=5, **kwargs):
         if not kwargs['auto_upload']:
             continue
 
+        if kwargs['auto_upload_minimal_size'] > get_file_size(output_file):
+            logger.info(f'auto_upload_minimal_size > get_file_size(output_file): {output_file}')
+            continue
+
         # move to upload folder
         dst_dir = os.path.join(upload_path, source_type, room_id)
         pathlib.Path(dst_dir).mkdir(parents=True, exist_ok=True)
@@ -166,6 +170,10 @@ def upload_validator(youtube):
     ).start()
 
 
+def get_file_size(path):
+    return '{:f}'.format(pathlib.Path(path).stat().st_size / 1024 / 1024)
+
+
 def main():
     config = get_config()
     youtube = recorder.destination.youtube.Youtube(config['youtube'])