fundrive-baidu 0.1.3__tar.gz

fundrive_baidu-0.1.3/PKG-INFO

@@ -0,0 +1,23 @@
+Metadata-Version: 2.1
+Name: fundrive-baidu
+Version: 0.1.3
+Summary: fundrive
+Author-email: niuliangtao <farfarfun@qq.com>
+Requires-Python: >=3.8
+Description-Content-Type: text/markdown
+Requires-Dist: aget>=0.2.0
+Requires-Dist: chardet>=5.2.0
+Requires-Dist: click>=8.1.7
+Requires-Dist: cryptography>=44.0.0
+Requires-Dist: cython>=3.0.11
+Requires-Dist: fastapi>=0.115.6
+Requires-Dist: jinja2>=3.1.4
+Requires-Dist: passlib>=1.7.4
+Requires-Dist: pillow>=10.4.0
+Requires-Dist: requests>=2.32.3
+Requires-Dist: requests-toolbelt>=1.0.0
+Requires-Dist: rich>=13.9.4
+Requires-Dist: typing-extensions>=4.12.2
+Requires-Dist: uvicorn>=0.33.0
+
+# fundrive-baidu

fundrive_baidu-0.1.3/README.md

@@ -0,0 +1 @@
+# fundrive-baidu

fundrive_baidu-0.1.3/pyproject.toml

@@ -0,0 +1,13 @@
+[project]
+name = "fundrive-baidu"
+version = "0.1.3"
+description = "fundrive"
+readme = "README.md"
+requires-python = ">=3.8"
+dependencies = [ "aget>=0.2.0", "chardet>=5.2.0", "click>=8.1.7", "cryptography>=44.0.0", "cython>=3.0.11", "fastapi>=0.115.6", "jinja2>=3.1.4", "passlib>=1.7.4", "pillow>=10.4.0", "requests>=2.32.3", "requests-toolbelt>=1.0.0", "rich>=13.9.4", "typing-extensions>=4.12.2", "uvicorn>=0.33.0",]
+[[project.authors]]
+name = "niuliangtao"
+email = "farfarfun@qq.com"
+
+[tool.setuptools]
+license-files = []

fundrive_baidu-0.1.3/setup.cfg

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

fundrive_baidu-0.1.3/src/baidu/__init__.py

@@ -0,0 +1,3 @@
+from .pcs import BaiduPCS, BaiduPCSApi
+
+__all__ = ["BaiduPCS", "BaiduPCSApi"]

fundrive_baidu-0.1.3/src/baidu/app/account.py

@@ -0,0 +1,223 @@
+import pickle
+from os import PathLike
+from pathlib import Path
+from typing import Dict, List, NamedTuple, Optional
+
+from rich import print
+
+from baidu.baidupcs import BaiduPCSApi, PcsUser
+from baidu.commands.env import ACCOUNT_DATA_PATH
+from baidu.common.path import join_path
+
+
+class Account(NamedTuple):
+    user: PcsUser
+
+    # current working directory
+    pwd: str = "/"
+    encrypt_password: Optional[str] = None
+
+    # In version 2, salt is generated by `os.urandom`.
+    # Here it useless.
+    salt: Optional[str] = None
+
+    # Account Name which can be set by hand
+    account_name: str = ""
+
+    def pcsapi(self) -> BaiduPCSApi:
+        auth = self.user.auth
+        assert auth, f"{self}.user.auth is None"
+        return BaiduPCSApi(
+            bduss=auth.bduss,
+            stoken=auth.stoken,
+            ptoken=auth.ptoken,
+            cookies=auth.cookies,
+            user_id=self.user.user_id,
+        )
+
+    @staticmethod
+    def from_bduss(
+        bduss: str,
+        cookies: Dict[str, Optional[str]] = {},
+        account_name: str = "",
+    ) -> "Account":
+        api = BaiduPCSApi(bduss=bduss, cookies=cookies)
+        user = api.user_info()
+        return Account(user=user, account_name=account_name or user.user_name or "")
+
+
+class AccountManager:
+    """Account Manager
+
+    Manage all accounts
+    """
+
+    def __init__(self, data_path: Optional[PathLike] = None):
+        self._accounts: Dict[int, Account] = {}  # user_id (int) -> Account
+        self._who: Optional[int] = None  # user_id (int)
+        self._data_path = data_path
+
+    @staticmethod
+    def load_data(data_path: PathLike) -> "AccountManager":
+        try:
+            data_path = Path(data_path).expanduser()
+            am = pickle.load(data_path.open("rb"))
+            _compat_account_manager(am)
+            return am
+        except Exception:
+            return AccountManager(data_path=data_path)
+
+    @property
+    def accounts(self) -> List[Account]:
+        """All accounts"""
+
+        return list(self._accounts.values())
+
+    def set_account_name(self, account_name: str, user_id: int = None):
+        """Set account name"""
+
+        user_id = user_id or self._who
+
+        assert user_id, "No recent user"
+
+        account = self._accounts.get(user_id)
+
+        assert account
+
+        account = account._replace(account_name=account_name)
+        self._accounts[user_id] = account
+
+    def set_encrypt_password(
+        self, encrypt_password: Optional[str] = None, salt: Optional[str] = None
+    ):
+        """Set encryption key"""
+
+        assert self._who, "No recent user"
+
+        account = self._accounts.get(self._who)
+
+        assert account
+
+        account = account._replace(encrypt_password=encrypt_password, salt=salt)
+        self._accounts[self._who] = account
+
+    def cd(self, remotedir: str = "/"):
+        """Change current working directory"""
+
+        assert self._who, "No recent user"
+
+        account = self._accounts.get(self._who)
+
+        assert account
+
+        pwd = join_path(Path(account.pwd), Path(remotedir))
+        account = account._replace(pwd=pwd)
+        self._accounts[self._who] = account
+
+    @property
+    def pwd(self) -> str:
+        """Current working directory of recent user"""
+
+        assert self._who, "No recent user"
+
+        account = self._accounts.get(self._who)
+
+        assert account
+
+        return account.pwd
+
+    def who(self, user_id: Optional[int] = None) -> Optional[Account]:
+        """Return recent `Account`"""
+
+        user_id = user_id or self._who
+        if user_id:
+            return self._accounts.get(user_id)
+        else:
+            return None
+
+    def update(self, user_id: Optional[int] = None):
+        """Update user_info"""
+
+        user_id = user_id or self._who
+        if user_id:
+            account = self._accounts.get(user_id)
+            if not account:
+                return None
+
+            api = account.pcsapi()
+            user = api.user_info()
+
+            self._accounts[user_id] = account._replace(user=user)
+
+    def su(self, user_id: int):
+        """Change recent user with `PcsUser.user_id`
+
+        Args:
+            who (int): `PcsUser.user_id`
+        """
+
+        assert user_id in self._accounts, f"No user {user_id}"
+
+        self._who = user_id
+
+    def add_account(self, account: Account):
+        """Add an account to the manager"""
+
+        self._accounts[account.user.user_id] = account
+
+    def delete_account(self, user_id: int):
+        """Delete a user
+
+        Args:
+            who (int): `PcsUser.user_id`
+        """
+
+        if user_id in self._accounts:
+            del self._accounts[user_id]
+        if user_id == self._who:
+            self._who = None
+
+    def save(self, data_path: Optional[PathLike] = None):
+        """Serialize to local path"""
+
+        data_path = data_path or self._data_path
+        assert data_path, "No data path"
+
+        data_path = Path(data_path).expanduser()
+        if not data_path.parent.exists():
+            data_path.parent.mkdir(parents=True)
+
+        pickle.dump(self, open(data_path, "wb"))
+
+
+def _compat_account_manager(am: AccountManager):
+    """Update stored account manager to compat current version"""
+    _compat_v0_5_9(am)
+    _set_account_names(am)
+
+
+def _compat_v0_5_9(am: AccountManager):
+    for user_id, account in am._accounts.items():
+        old_products = account.user.products
+        if isinstance(old_products, dict):  # type: ignore
+            print(  # type: ignore
+                "[i yellow]Update[/i yellow]: Transfer PcsUser format for ^v0.5.9: "
+                f"user_id: {user_id}"
+            )
+            am.update(user_id)
+
+    data_path = am._data_path
+    if not data_path or not Path(data_path).exists():
+        am._data_path = ACCOUNT_DATA_PATH
+
+    am.save()
+
+
+def _set_account_names(am: AccountManager):
+    """Set account's name with its user's name if that is empty"""
+
+    for user_id, account in am._accounts.items():
+        if not account.account_name:
+            am._accounts[user_id] = account._replace(
+                account_name=account.user.user_name or ""
+            )