optimize: clean up code

ayugespidertools/commands/startproject.py

@@ -1,70 +1,16 @@
-import string
+import argparse
 from pathlib import Path
-from shutil import ignore_patterns, move
+from typing import List
 
 from scrapy.commands.startproject import Command
-from scrapy.exceptions import UsageError
-from scrapy.utils.template import render_templatefile, string_camelcase
 
 import ayugespidertools
 
-# 添加需要的自定义配置文件
-TEMPLATES_TO_RENDER = (
-    ("scrapy.cfg",),
-    ("${project_name}", "settings.py.tmpl"),
-    ("${project_name}", "items.py.tmpl"),
-    ("${project_name}", "pipelines.py.tmpl"),
-    ("${project_name}", "middlewares.py.tmpl"),
-)
-
-IGNORE = ignore_patterns("*.pyc", "__pycache__", ".svn")
-
 
 class AyuCommand(Command):
-    def run(self, args, opts):
-        if len(args) not in (1, 2):
-            raise UsageError()
-
-        project_name = args[0]
-
-        if len(args) == 2:
-            project_dir = Path(args[1])
-        else:
-            project_dir = Path(args[0])
-
-        if (project_dir / "scrapy.cfg").exists():
-            self.exitcode = 1
-            print(f"Error: scrapy.cfg already exists in {project_dir.resolve()}")
-            return
-
-        if not self._is_valid_name(project_name):
-            self.exitcode = 1
-            return
-
-        self._copytree(Path(self.templates_dir), project_dir.resolve())
-        move(project_dir / "module", project_dir / project_name)
-        for paths in TEMPLATES_TO_RENDER:
-            tplfile = Path(
-                project_dir,
-                *(
-                    string.Template(s).substitute(project_name=project_name)
-                    for s in paths
-                ),
-            )
-            render_templatefile(
-                tplfile,
-                project_name=project_name,
-                ProjectName=string_camelcase(project_name),
-            )
-
-        print(
-            f"New Scrapy project '{project_name}', using template directory "
-            f"'{self.templates_dir}', created in:"
-        )
-        print(f"    {project_dir.resolve()}\n")
-        print("You can start your first spider with:")
-        print(f"    cd {project_dir}")
-        print("    scrapy genspider example example.com")
+    def run(self, args: List[str], opts: argparse.Namespace) -> None:
+        # 若想自定义 TEMPLATES_TO_RENDER 的文件模版，需重写父类的 run 方法，示例请查看提交历史
+        super(AyuCommand, self).run(args, opts)
         # 添加本库的文字提示内容
         print("Or you can start your first spider with ayuge:")
         print("    ayuge genspider example example.com")