fiuai-sdk-python 0.2.4__tar.gz → 0.2.6__tar.gz

{fiuai_sdk_python-0.2.4 → fiuai_sdk_python-0.2.6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: fiuai_sdk_python
-Version: 0.2.4
+Version: 0.2.6
 Summary: FiuAI Python SDK - 企业级AI服务集成开发工具包
 Project-URL: Homepage, https://github.com/fiuai/fiuai-sdk-python
 Project-URL: Documentation, https://github.com/fiuai/fiuai-sdk-python#readme

{fiuai_sdk_python-0.2.4 → fiuai_sdk_python-0.2.6}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "fiuai_sdk_python"
-version = "0.2.4"
+version = "0.2.6"
 description = "FiuAI Python SDK - 企业级AI服务集成开发工具包"
 readme = "README.md"
 requires-python = ">=3.12"

{fiuai_sdk_python-0.2.4 → fiuai_sdk_python-0.2.6}/src/fiuai_sdk_python/bank.py

@@ -110,30 +110,30 @@ def get_bank_account_by_bank_id(client: FiuaiSDK, bank_id: str) -> BankAccount|N
 
     # 获取支行信息
     bank_branch = ""
-    if resp[0]["opening_bank_branch"] != "":
+    if resp["opening_bank_branch"] != "":
         _bank_branch = client.get_doc(
             doctype="Bank Branch",
-            name=resp[0]["opening_bank_branch"],
+            name=resp["opening_bank_branch"],
             fields=["name", "bank_branch_name"]
         )
         if not _bank_branch:
-            logger.error(f"bank branch is not found by id {resp[0]["opening_bank_branch"]}")
+            logger.error(f"bank branch is not found by id {resp["opening_bank_branch"]}")
             return None
         else:
-            bank_branch = _bank_branch[0]["bank_branch_name"]
+            bank_branch = _bank_branch["bank_branch_name"]
 
     # 获取银行信息
     _bank_detail = client.get_doc(
         doctype="Bank",
-        name=resp[0]["bank"],
+        name=resp["bank"],
         fields=["name", "bank_name"]
     )
 
     if not _bank_detail:
-        logger.error(f"bank is not found by id {resp[0]["bank"]}")
+        logger.error(f"bank is not found by id {resp["bank"]}")
         return None
 
-    _bank_account = resp[0]
+    _bank_account = resp
     return BankAccount(
         name=_bank_account["name"],
         account_name=_bank_account.get("account_name", "") if _bank_account.get("account_name", "") else "",

{fiuai_sdk_python-0.2.4 → fiuai_sdk_python-0.2.6}/src/fiuai_sdk_python/client.py

@@ -77,7 +77,9 @@ class FiuaiSDK(object):
     
     def _logout(self):
         logger.info(f"Logout from {self.url}")
-        self.client.get(self.url, params={"cmd": "logout"}, headers=self.headers)
+        if self.auth_type == "password":
+            # internal login 不需要logout
+            self.client.get(self.url, params={"cmd": "logout"}, headers=self.headers)
 
 
     def __enter__(self):
@@ -205,7 +207,12 @@ class FiuaiSDK(object):
         res = self.client.get(self.url + '/api/resource/' + doctype + '/' + name,
                                 params=params, headers=self.headers)
 
-        return self.post_process(res)
+        
+        docs = self.post_process(res)
+        if isinstance(docs, list):
+            return docs[0]
+        else:
+            return docs
 
 
     def get_api(self, method, params={}):

{fiuai_sdk_python-0.2.4 → fiuai_sdk_python-0.2.6}/src/fiuai_sdk_python/company.py

@@ -44,7 +44,6 @@ def get_target_company_by_name(client: FiuaiSDK, full_name: str, target_side: Li
         logger.debug(f"get_target_company_by_name: {full_name} not found")
         return TargetCompanySearchResult(name="", full_name="", unique_no="", exists=False, in_network=False, comment="该公司不存在")
     
-    _lookup = _lookup[0]
 
     _network = client.get_doc(
         doctype="Company Network", 
@@ -54,8 +53,6 @@ def get_target_company_by_name(client: FiuaiSDK, full_name: str, target_side: Li
         logger.debug(f"get_target_company_by_name: {full_name} is not in network")
         return TargetCompanySearchResult(name="", full_name="", unique_no="", exists=True, in_network=False, comment="该公司不在本方网络中")
     
-    _network = _network[0]
-
 
     n = TargetCompanySearchResult(
             name=_network["target_company_id"], 
@@ -98,7 +95,7 @@ def get_company_profile(client: FiuaiSDK, auth_company_id: str)-> CompanyProfile
     if not resp:
         return None
 
-    r = resp[0]
+    r = resp
    
     if r["default_bank_account"]:
         rr = get_bank_account_by_bank_id(client, r["default_bank_account"])

fiuai_sdk_python-0.2.6/src/fiuai_sdk_python/perm.py

@@ -0,0 +1,8 @@
+# -- coding: utf-8 --
+# Project: fiuai_sdk_python
+# Created Date: 2025 09 Th
+# Author: liming
+# Email: lmlala@aliyun.com
+# Copyright (c) 2025 FiuAI
+
+

{fiuai_sdk_python-0.2.4 → fiuai_sdk_python-0.2.6}/src/fiuai_sdk_python/type.py

@@ -12,7 +12,21 @@ from datetime import datetime
 import json
 import re
 from .token import TokenConfig
+from enum import StrEnum
 
+
+########## client type ##########
+
+
+class ClientConfig(BaseModel):
+    url: str
+    max_api_retry: int
+    timeout: int
+    verify: bool
+    tokens: List[TokenConfig] = []
+
+
+########## user type ##########
 class UserProfile(BaseModel):
     name: str = Field(description="用户名称")
     full_name: str = Field(description="用户全名")
@@ -22,6 +36,36 @@ class UserProfile(BaseModel):
     default_company_id: str = Field(description="用户默认公司", default="")
     phone: str = Field(description="用户电话", default="")
 
+class DocTypePermission(StrEnum):
+    SELECT = "can_select"
+    READ = "can_read"
+    WRITE = "can_write"
+    CREATE = "can_create"
+    DELETE = "can_delete"
+    SUBMIT = "can_submit"
+    CANCEL = "can_cancel"
+    AMEND = "can_amend"
+    APPROVE = "can_approve"
+    EXPORT = "can_export"
+    IMPORT = "can_import"
+
+class UserDocTypePermission(BaseModel):
+    can_select: bool
+    can_read: bool
+    can_write: bool
+    can_create: bool
+    can_delete: bool
+    can_submit: bool
+    can_cancel: bool
+    can_amend: bool
+    can_approve: bool
+    can_export: bool
+    can_import: bool
+
+
+
+########## basic ##########
+
 
 class ClientConfig(BaseModel):
     url: str