donlad-stats 0.1.0__tar.gz

donlad_stats-0.1.0/PKG-INFO

@@ -0,0 +1,11 @@
+Metadata-Version: 2.4
+Name: donlad_stats
+Version: 0.1.0
+Summary: donlad 下载量统计工具（SQLite + JSON + TXT）
+Author: shiroko
+Requires-Python: >=3.6
+Requires-Dist: donlad
+Dynamic: author
+Dynamic: requires-dist
+Dynamic: requires-python
+Dynamic: summary

donlad_stats-0.1.0/donlad_stats/__init__.py

@@ -0,0 +1,5 @@
+from .core import Stats
+from .cli import main
+
+__version__ = "0.1.0"
+__all__ = ["Stats", "main"]

donlad_stats-0.1.0/donlad_stats/cli.py

@@ -0,0 +1,29 @@
+#!/usr/bin/env python3
+import argparse
+from donlad.cli import load_config
+from .core import Stats
+
+def main():
+    parser = argparse.ArgumentParser(description="donlad 统计工具")
+    parser.add_argument("--save", action="store_true", help="查询所有已添加模块并保存")
+    parser.add_argument("--packages", nargs="+", help="临时指定模块列表")
+    args = parser.parse_args()
+
+    if args.save:
+        config = load_config()
+        pkgs = config.get("packages", [])
+        if not pkgs:
+            print("暂无模块，请先用 donlad add 添加")
+            return
+        stats = Stats(pkgs)
+        stats.run_all()
+        stats.close()
+    elif args.packages:
+        stats = Stats(args.packages)
+        stats.run_all()
+        stats.close()
+    else:
+        parser.print_help()
+
+if __name__ == "__main__":
+    main()

donlad_stats-0.1.0/donlad_stats/core.py

@@ -0,0 +1,74 @@
+import time
+import secrets
+import sqlite3
+import os
+import json
+from datetime import datetime
+from donlad import get_downloads_robust as get_downloads
+
+DEFAULT_CONFIG_DIR = os.path.expanduser("~/.donlad_stats")
+DB_PATH = os.path.join(DEFAULT_CONFIG_DIR, "downloads.db")
+JSON_PATH = os.path.join(DEFAULT_CONFIG_DIR, "downloads.json")
+TXT_PATH = os.path.join(DEFAULT_CONFIG_DIR, "downloads.txt")
+
+os.makedirs(DEFAULT_CONFIG_DIR, exist_ok=True)
+
+def randint(a, b):
+    return a + secrets.randbelow(b - a + 1)
+
+class Stats:
+    def __init__(self, packages=None):
+        self.packages = packages or []
+        self.conn = sqlite3.connect(DB_PATH)
+        self._init_db()
+        self.now = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
+
+    def _init_db(self):
+        c = self.conn.cursor()
+        c.execute("""
+            CREATE TABLE IF NOT EXISTS stats (
+                name TEXT PRIMARY KEY,
+                downloads INTEGER,
+                time TEXT
+            )
+        """)
+        self.conn.commit()
+
+    def _update_now(self):
+        self.now = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
+
+    def fetch(self, pkg):
+        self._update_now()
+        downloads = get_downloads(pkg)
+        c = self.conn.cursor()
+        c.execute("INSERT OR REPLACE INTO stats VALUES (?, ?, ?)", (pkg, downloads, self.now))
+        self.conn.commit()
+        return downloads
+
+    def save_json(self):
+        data = {}
+        c = self.conn.cursor()
+        c.execute("SELECT name, downloads, time FROM stats")
+        for name, downloads, t in c.fetchall():
+            data[name] = {"downloads": downloads, "time": t}
+        with open(JSON_PATH, "w", encoding="utf-8") as f:
+            json.dump(data, f, ensure_ascii=False, indent=2)
+
+    def save_txt(self):
+        c = self.conn.cursor()
+        c.execute("SELECT name, downloads, time FROM stats")
+        rows = c.fetchall()
+        with open(TXT_PATH, "w", encoding="utf-8") as f:
+            for name, downloads, t in rows:
+                f.write(f"name:{name}\ndownloads:{downloads}\ntime:{t}\n\n")
+
+    def run_all(self, delay_range=(1, 3)):
+        for pkg in self.packages:
+            time.sleep(randint(*delay_range))
+            downloads = self.fetch(pkg)
+            print(f"['name:{pkg}']\n['{pkg} download:{downloads}']\n")
+        self.save_json()
+        self.save_txt()
+
+    def close(self):
+        self.conn.close()

donlad_stats-0.1.0/donlad_stats.egg-info/PKG-INFO

@@ -0,0 +1,11 @@
+Metadata-Version: 2.4
+Name: donlad_stats
+Version: 0.1.0
+Summary: donlad 下载量统计工具（SQLite + JSON + TXT）
+Author: shiroko
+Requires-Python: >=3.6
+Requires-Dist: donlad
+Dynamic: author
+Dynamic: requires-dist
+Dynamic: requires-python
+Dynamic: summary

donlad_stats-0.1.0/donlad_stats.egg-info/SOURCES.txt

@@ -0,0 +1,11 @@
+README.md
+setup.py
+donlad_stats/__init__.py
+donlad_stats/cli.py
+donlad_stats/core.py
+donlad_stats.egg-info/PKG-INFO
+donlad_stats.egg-info/SOURCES.txt
+donlad_stats.egg-info/dependency_links.txt
+donlad_stats.egg-info/entry_points.txt
+donlad_stats.egg-info/requires.txt
+donlad_stats.egg-info/top_level.txt

donlad_stats-0.1.0/donlad_stats.egg-info/dependency_links.txt

@@ -0,0 +1 @@
+

donlad_stats-0.1.0/donlad_stats.egg-info/entry_points.txt

@@ -0,0 +1,2 @@
+[console_scripts]
+donlad-stats = donlad_stats.cli:main

donlad_stats-0.1.0/donlad_stats.egg-info/requires.txt

@@ -0,0 +1 @@
+donlad

donlad_stats-0.1.0/donlad_stats.egg-info/top_level.txt

@@ -0,0 +1 @@
+donlad_stats

donlad_stats-0.1.0/setup.cfg

@@ -0,0 +1,4 @@
+[egg_info]
+tag_build = 
+tag_date = 0
+

donlad_stats-0.1.0/setup.py

@@ -0,0 +1,17 @@
+from setuptools import setup, find_packages
+
+setup(
+    name="donlad_stats",
+    version="0.1.0",
+    author="shiroko",
+    description="donlad 下载量统计工具（SQLite + JSON + TXT）",
+    packages=find_packages(),
+    install_requires=["donlad"],
+    entry_points={
+        "console_scripts": [
+            "donlad-stats = donlad_stats.cli:main",
+        ]
+    },
+    python_requires=">=3.6",
+)
+