💅重构Command.py #155 #191 #202 #203 #206 #220 #226 #227 #243 #267 #287 #…

…294 #312 #320 #332 #335 #341 #343 #346 #355 #357 #360 #361 #362 #364 #365 #367 #370 #372 #374 #376 #377 #379 #380 #381 #382 #383 #393 #394 #397 #398 #399 #401 #402 #403 #404 #406 #407 #408 #410 #414 #416 #419 #420 #422 #423 #426 #431 #432 #433 #434 #435 #436 #437 #446 #448 #449 #450 #451 #452 #454 #455 #456 #567 #460 #466 #469 #470 #471 #472 #475 #476 #477 #480 #481 #483 #484 #485

1、修改命令行选项
2、添加全局headers
#155 #191 #202 #203 #206 #220 #226 #227 #243 #267 #287 #294 #312 #320 #332 #335 #341 #343 #346 #355 #357 #360 #361 #362 #364 #365 #367 #370 #372 #374 #376 #377 #379 #380 #381 #382 #383 #393 #394 #397 #398 #399 #401 #402 #403 #404 #406 #407 #408 #410 #414 #416 #419 #420 #422 #423 #426 #431 #432 #433 #434 #435 #436 #437 #446 #448 #449 #450 #451 #452 #454 #455 #456 #567 #460 #466 #469 #470 #471 #472 #475 #476 #477 #480 #481 #483 #484 #485

Util/Command.py

@@ -21,10 +21,17 @@
 class Command:
 
     def __init__(self):
-        # 初始化配置文件
-        self.cfgs = Util.Config()
-        # 检查配置文件是否存在
-        self.cfg = self.cfgs.check()
+        # 字典类型配置文件
+        self.config_dict = {}
+        # 全局headers
+        self.dyheaders = {
+            'Cookie': '',
+            'User-Agent':'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/104.0.0.0 Safari/537.36',
+            'Referer':'https://www.douyin.com/'
+        }
+        self.setting()
+        # 是否自动升级
+        Util.Updata(self.config_dict['update'])
 
     def argument(self):
         """
@@ -33,18 +40,18 @@ def argument(self):
             args: 返回命令行对象
         """
         parser = Util.argparse.ArgumentParser(
-            description='TikTokTool V1.3.0.90 使用帮助')
+            description='TikTokTool V1.4.0.0 使用帮助')
         parser.add_argument('--uid', '-u', type=str,
                             help='为用户主页链接，支持长短链', required=False)
-        # parser.add_argument('--dir','-d', type=str,help='视频保存目录，非必要参数， 默认./Download', default='./Download/')
+        parser.add_argument('--save','-s', type=str,help='视频保存目录，非必要参数， 默认Download', default='Download')
         # parser.add_argument('--single', '-s', type=str, help='单条视频链接，非必要参数，与--user参数冲突')
         parser.add_argument('--music', '-m', type=str,
                             help='是否下载视频原声， 默认no 可选yes', default='no')
         # parser.add_argument('--count', '-c', type=int, help='单页下载的数量，默认参数 35 无须修改', default=35)
         parser.add_argument('--mode', '-M', type=str,
                             help='下载模式选择，默认post:发布的视频 like:点赞视频(需要开放权限) collection:收藏夹视频(需要登录账号详情看cookie帮助)', default='post')
         parser.add_argument('--cookie', '-cookie', type=str,
-                            help='请求接口需要cookie，个性化推荐内容填入sessionid_ss，其他填入odin_tt、passport_csrf_token', default='')
+                            help='请求大部分接口需要cookie，请调用扫码登录填写cookie', default='', required=False)
         parser.add_argument('--interval', '-I', type=str,
                             help='根据作品发布日期区间下载作品，例如2022-01-01|2023-01-01下载的是2022年所有作品，0为下载全部', default='0')
         parser.add_argument('--update', '-U', type=str,
@@ -56,31 +63,27 @@ def setting(self):
         """
         设置配置
         Returns:
-            tuple: 返回uid,music,mode,cookie,interval,update
+            dict: 返回字典类型配置文件
         """
-        args = self.argument()
-        # 检测是否为命令行调用
-        if args.uid == None:
-            print('[  警告  ]:未检测到命令，将使用配置文件进行批量下载!')
-            self.uid = self.cfg.get('uid', 'uid')
-            self.music = self.cfg.get('music', 'music')
-            self.mode = self.cfg.get('mode', 'mode')
-            self.cookie = self.cfg.get('cookie', 'cookie')
-            self.interval = self.cfg.get('interval', 'interval')
-            self.update = self.cfg.get('update', 'update')
-            print('[  提示  ]:读取本地配置完成!\r')
-            Util.log.info('[  提示  ]:读取本地配置完成!')
-        else:
-            self.uid = args.uid
-            self.music = args.music
-            self.mode = args.mode
-            self.cookie = args.cookie
-            self.interval = args.interval
-            self.update = args.update
-            print('[  提示  ]:读取命令完成!\r')
-            Util.log.info('[  提示  ]:读取命令完成!')
-        return [self.uid, self.music, self.mode, self.cookie, self.interval, self.update]
 
+        # 检查配置文件是否存在
+        cfg = Util.Config().check()
+        if cfg['cookie'] == '':
+            # sso登录
+            login = Util.Login()
+            self.dyheaders = login.loginHeaders
+        else:
+            self.dyheaders['Cookie'] = cfg['cookie']
 
-if __name__ == '__main__':
-    Command()
+        # 检测是否为命令行调用
+        args = self.argument()
+        # 如果args中有任何非None的值则设置为命令行
+        if len(Util.sys.argv) > 1:  # 如果命令行参数列表的长度大于1，说明有提供命令行参数
+            args = self.argument()
+            self.config_dict = vars(args)
+            Util.console.print('[  配置  ]:获取命令行完成!\r')
+            Util.log.info('[  配置  ]:获取命令行完成!')
+        else:
+            self.config_dict = cfg
+            Util.console.print('[  配置  ]:读取本地配置完成!\r')
+            Util.log.info('[  配置  ]:读取本地配置完成!')