add:添加资源同步进度条

README.md

@@ -45,9 +45,12 @@
 
 # 2. 版本号
 
-1.5.1
+1.6.0
 
 ## 2.1 更新内容
+- [1.6.0]  
+    > 文件同步添加了进度条，方便查看文件同步信息，但请注意，若下载过程中未完成中断任务需将下载一半的文件删除，否则不会更新
+    ![](img/1.6.0.png)
 - [1.5.1]  
     > 修复下载全部课程资源时没有打开文件目录的选项问题，当无更新的时候自动退出应用
     

core/source.py

@@ -13,6 +13,8 @@
 
 from bs4 import BeautifulSoup
 
+from core.utils import download_file
+
 sys.setrecursionlimit(100000)
 
 
@@ -211,9 +213,10 @@ def _download_one(self, course_info, source_info):
         if not os.path.isfile(file_path):
             # 只下载没有的文件，若存在不下载，节省流量
             self._logger.info("正在下载:{}".format(source_info['name']))
-            content = self._S.get(source_info['url']).content
-            with open(file_path, 'wb') as f:
-                f.write(content)
+            # content = self._S.get(source_info['url']).content
+            # with open(file_path, 'wb') as f:
+            #     f.write(content)
+            download_file(url=source_info['url'], session=self._S, file_path=file_path)
             self._update_sources.append("[{}:{}]".format(course_info["name"], source_info['name']))  # 记录更新的课程数据
 
     def _download_course(self, course_info):

core/utils.py

@@ -5,6 +5,9 @@
 # @Time    : 2019/11/28/028 14:11
 # @WebSite : https://www.gentlecp.com
 import requests
+import os
+import logging
+from tqdm import tqdm
 from urllib.parse import urlparse
 
 def util_login(stuid,password):
@@ -33,3 +36,34 @@ def util_login(stuid,password):
         }
 
 
+def download_file(url, session=None, file_path='未命名文件', overwrite = False):
+    '''
+    根据指定url下载文件
+    :param url:
+    :param session: 传入的会话参数，有的需要登录才能下载
+    :param file_path: 文件存储路径，默认为当前目录下，存储文件为未命名文件
+    :param overwrite: 是否覆盖同名文件，默认否
+    :return: 正确下载返回True，否则False
+    '''
+    if session:
+        res = session.get(url, stream = True)
+    else:
+        res = requests.get(url,stream=True)
+    file_size = int(res.headers['content-length'])
+    chunk_size = 1024
+    if res.status_code == 200:
+        if not overwrite and os.path.exists(file_path):
+            return True
+        else:
+            progress_bar = tqdm(
+                total=file_size,initial=0,unit='B',unit_scale=True,
+            )
+            with open(file_path,'wb') as f:
+                for data in res.iter_content(chunk_size=chunk_size):
+                    if data:
+                        f.write(data)
+                        progress_bar.update(chunk_size)
+            progress_bar.close()
+    else:
+        logging.error('download fail.')
+        return False

environment.yml

@@ -29,6 +29,7 @@ dependencies:
     - prettytable==0.7.2
     - requests==2.22.0
     - soupsieve==1.9.5
+    - tqdm==4.45.0
     - urllib3==1.25.7
 prefix: /Users/dongchaopeng/miniconda3/envs/UCASHelper
 

requirements.txt

@@ -5,4 +5,5 @@ idna==2.8
 prettytable==0.7.2
 requests==2.22.0
 soupsieve==1.9.5
+tqdm==4.45.0
 urllib3==1.25.7