Refactor the code to make it comply with PEP-8 standards:Convert all class definitions to CamelCase style.
(issue 37)
This commit is contained in:
@@ -1,3 +1,3 @@
|
||||
from .agent_manager import agent_manager,agent_manager_client
|
||||
from .agent_manager import AgentManager,AgentManagerClient
|
||||
from .agent import ai_agent
|
||||
from .templete import ai_agent_templete
|
||||
@@ -2,14 +2,13 @@
|
||||
import logging
|
||||
import toml
|
||||
|
||||
from .agent import ai_agent
|
||||
from .templete import ai_agent_templete
|
||||
from package_manager import pkg_env,pkg_env_mgr,pkg_media_info,media_reader,pkg_install_task
|
||||
from aios_kernel import AIAgent,AIAgentTemplete
|
||||
from package_manager import PackageEnv,PackageEnvManager,PackageMediaInfo,PackageInstallTask
|
||||
|
||||
logger = logging.getLogger(__name__)
|
||||
|
||||
|
||||
class agent_manager:
|
||||
class AgentManager:
|
||||
_instance = None
|
||||
|
||||
def __init__(self) -> None:
|
||||
@@ -18,18 +17,18 @@ class agent_manager:
|
||||
|
||||
def __new__(cls):
|
||||
if cls._instance is None:
|
||||
cls._instance = super(agent_manager, cls).__new__(cls)
|
||||
cls._instance = super(AgentManager, cls).__new__(cls)
|
||||
return cls._instance
|
||||
|
||||
|
||||
def initial(self,root_dir:str) -> None:
|
||||
self.agent_templete_env : pkg_env = pkg_env_mgr().get_env(f"{root_dir}templetes/agent_templetes.cfg")
|
||||
self.agent_env : pkg_env = pkg_env_mgr().get_env(f"{root_dir}agents/agents.cfg")
|
||||
self.agent_templete_env : PackageEnv = PackageEnvManager().get_env(f"{root_dir}templetes/agent_templetes.cfg")
|
||||
self.agent_env : PackageEnv = PackageEnvManager().get_env(f"{root_dir}agents/agents.cfg")
|
||||
if self.agent_templete_env is None:
|
||||
raise Exception("agent_manager initial failed")
|
||||
|
||||
|
||||
def get(self,agent_id:str) -> ai_agent:
|
||||
def get(self,agent_id:str) -> AIAgent:
|
||||
the_agent = self.loaded_agent_instance.get(agent_id)
|
||||
if the_agent:
|
||||
return the_agent
|
||||
@@ -39,7 +38,7 @@ class agent_manager:
|
||||
if agent_media_info is None:
|
||||
return None
|
||||
|
||||
the_agent = self._load_agent_from_media(agent_media_info)
|
||||
the_agent : AIAgent = self._load_agent_from_media(agent_media_info)
|
||||
if the_agent is None:
|
||||
logger.warn(f"load agent {agent_id} from media failed!")
|
||||
|
||||
@@ -49,29 +48,25 @@ class agent_manager:
|
||||
def remove(self,agent_id:str)->int:
|
||||
pass
|
||||
|
||||
def get_templete(self,templete_id) -> ai_agent_templete:
|
||||
def get_templete(self,templete_id) -> AIAgentTemplete:
|
||||
template_media_info = self.agent_templete_env.get(templete_id)
|
||||
if template_media_info is None:
|
||||
return None
|
||||
return self._load_templete_from_media(template_media_info)
|
||||
|
||||
def install(self,templete_id) -> pkg_install_task:
|
||||
def install(self,templete_id) -> PackageInstallTask:
|
||||
installer = self.agent_templete_env.get_installer()
|
||||
return installer.install(templete_id)
|
||||
|
||||
def uninstall(self,templete_id) -> int:
|
||||
pass
|
||||
|
||||
def _load_templete_from_media(self,templete_media:pkg_media_info) -> ai_agent_templete:
|
||||
def _load_templete_from_media(self,templete_media:PackageMediaInfo) -> AIAgentTemplete:
|
||||
pass
|
||||
|
||||
def _load_agent_from_media(self,agent_media:pkg_media_info) -> ai_agent:
|
||||
def _load_agent_from_media(self,agent_media:PackageMediaInfo) -> AIAgent:
|
||||
pass
|
||||
|
||||
def create(self,template,agent_name,agent_last_name,agent_introduce) -> ai_agent:
|
||||
def create(self,template,agent_name,agent_last_name,agent_introduce) -> AIAgent:
|
||||
pass
|
||||
|
||||
|
||||
class agent_manager_client:
|
||||
def __init__(self) -> None:
|
||||
pass
|
||||
@@ -1 +1,2 @@
|
||||
from .cid import content_id
|
||||
from .cid import ContentId
|
||||
from .ndn_client import NDN_Client
|
||||
@@ -1,8 +1,9 @@
|
||||
|
||||
|
||||
class content_id:
|
||||
class ContentId:
|
||||
def __init__(self) -> None:
|
||||
pass
|
||||
|
||||
@staticmethod
|
||||
def create_from_str(cid_str:str):
|
||||
pass
|
||||
@@ -1,13 +1,13 @@
|
||||
|
||||
from .cid import content_id
|
||||
from .cid import ContentId
|
||||
|
||||
class ndn_get_task:
|
||||
class NDN_GetTask:
|
||||
def __init__(self) -> None:
|
||||
pass
|
||||
|
||||
class ndn_client:
|
||||
class NDN_Client:
|
||||
def __init__(self):
|
||||
pass
|
||||
|
||||
def get_file(cid:content_id,target_path:str,urls:{}=None,options:{}=None):
|
||||
def get_file(self,cid:ContentId,target_path:str,urls:{}=None,options:{}=None):
|
||||
pass
|
||||
@@ -1,3 +1,3 @@
|
||||
from .env import pkg_env_mgr,pkg_env
|
||||
from .pkg import pkg_info,pkg_media_info
|
||||
from .installer import pkg_install_task
|
||||
from .env import PackageEnvManager,PackageEnv
|
||||
from .pkg import PackageInfo,PackageMediaInfo
|
||||
from .installer import PackageInstallTask
|
||||
@@ -2,16 +2,16 @@
|
||||
import logging
|
||||
import toml
|
||||
|
||||
from .pkg import pkg_info,pkg_media_info
|
||||
from .installer import pkg_installer
|
||||
from .pkg import PackageInfo,PackageMediaInfo
|
||||
from .installer import PackageInstaller
|
||||
|
||||
logger = logging.getLogger(__name__)
|
||||
|
||||
class pkg_env_mgr:
|
||||
class PackageEnvManager:
|
||||
_instance = None
|
||||
def __new__(cls):
|
||||
if cls._instance is None:
|
||||
cls._instance = super(pkg_env_mgr, cls).__new__(cls)
|
||||
cls._instance = super(PackageEnvManager, cls).__new__(cls)
|
||||
return cls._instance
|
||||
|
||||
def __init__(self) -> None:
|
||||
@@ -19,46 +19,46 @@ class pkg_env_mgr:
|
||||
|
||||
pass
|
||||
|
||||
def get_env(self,cfg_path:str) -> pkg_env:
|
||||
def get_env(self,cfg_path:str) -> PackageEnv:
|
||||
if cfg_path in self._pkg_envs:
|
||||
return self._pkg_envs[cfg_path]
|
||||
else:
|
||||
pkg_env = pkg_env(cfg_path)
|
||||
pkg_env = PackageEnv(cfg_path)
|
||||
self._pkg_envs[cfg_path] = pkg_env
|
||||
return pkg_env
|
||||
|
||||
def get_user_env(self) -> pkg_env:
|
||||
def get_user_env(self) -> PackageEnv:
|
||||
pass
|
||||
|
||||
def get_system_env(self) -> pkg_env:
|
||||
def get_system_env(self) -> PackageEnv:
|
||||
pass
|
||||
|
||||
class pkg_env:
|
||||
class PackageEnv:
|
||||
def __init__(self,cfg_path:str) -> None:
|
||||
self.pkg_dir : str = ""
|
||||
self.pkg_obj_dir : str = ""
|
||||
self.is_strict : bool = True
|
||||
self.parent_envs : list[pkg_env] = None
|
||||
self.parent_envs : list[PackageEnv] = None
|
||||
self.index_dbs = None
|
||||
|
||||
self.cfg_path = cfg_path
|
||||
self._load_pkg_cfg(cfg_path)
|
||||
pass
|
||||
|
||||
def load(self,pkg_name:str,search_parent=True) -> pkg_media_info:
|
||||
def load(self,pkg_name:str,search_parent=True) -> PackageMediaInfo:
|
||||
pkg_path = None
|
||||
pkg_id,verion_str,cid = pkg_info.parse_pkg_name(pkg_name)
|
||||
pkg_id,verion_str,cid = PackageInfo.parse_pkg_name(pkg_name)
|
||||
|
||||
if cid is None:
|
||||
if verion_str is None:
|
||||
channel = self.get_pkg_channel_from_version(verion_str)
|
||||
channel:str = self.get_pkg_channel_from_version(verion_str)
|
||||
if channel is None:
|
||||
pkg_path = f"{self.pkg_dir}{pkg_id}"
|
||||
else:
|
||||
pkg_path = f"{self.pkg_dir}{pkg_id}#{channel}"
|
||||
else:
|
||||
channel = self.get_pkg_channel_from_version(verion_str)
|
||||
the_version = self.get_exact_version_from_installed(verion_str)
|
||||
channel:str = self.get_pkg_channel_from_version(verion_str)
|
||||
the_version:str = self.get_exact_version_from_installed(verion_str)
|
||||
if the_version is None:
|
||||
logger.warn(f"load {pkg_name} failed: no match version from {verion_str}")
|
||||
return None
|
||||
@@ -69,7 +69,7 @@ class pkg_env:
|
||||
else:
|
||||
pkg_path = f"{self.pkg_obj_dir}.{pkg_id}/{cid}"
|
||||
|
||||
media_info = self.try_load_pkg_media_info(pkg_id,pkg_path)
|
||||
media_info:PackageMediaInfo = self.try_load_pkg_media_info(pkg_path)
|
||||
if media_info is None:
|
||||
if search_parent:
|
||||
for parent_env in self.parent_envs:
|
||||
@@ -86,28 +86,28 @@ class pkg_env:
|
||||
def get_pkg_channel_from_version(self,pkg_version:str) -> str:
|
||||
pass
|
||||
|
||||
def get_pkg_media_info(self,pkg_name:str)->pkg_media_info:
|
||||
def get_pkg_media_info(self,pkg_name:str)->PackageMediaInfo:
|
||||
pass
|
||||
|
||||
def try_load_pkg_media_info(self,pkg_full_path:str) -> pkg_media_info:
|
||||
def try_load_pkg_media_info(self,pkg_full_path:str) -> PackageMediaInfo:
|
||||
pass
|
||||
|
||||
|
||||
def get_installed_pkg_info(self,pkg_name:str) -> pkg_info:
|
||||
def get_installed_pkg_info(self,pkg_name:str) -> PackageInfo:
|
||||
pass
|
||||
|
||||
def lookup(self,pkg_id:str,version_str:str) -> pkg_info:
|
||||
def lookup(self,pkg_id:str,version_str:str) -> PackageInfo:
|
||||
# to make sure pkg.cid is correct, we MUST verfiy eveything here
|
||||
pass
|
||||
|
||||
def get_installer(self) -> pkg_installer:
|
||||
def get_installer(self) -> PackageInstaller:
|
||||
pass
|
||||
|
||||
@classmethod
|
||||
def is_valied_media(pkg_full_path:str) -> bool:
|
||||
pass
|
||||
|
||||
def do_pkg_media_trans(self,pkg_info:pkg_info,source_path:str,target_path:str) -> bool:
|
||||
def do_pkg_media_trans(self,pkg_info:PackageInfo,source_path:str,target_path:str) -> bool:
|
||||
pass
|
||||
|
||||
def _load_pkg_cfg(self,cfg_path:str):
|
||||
|
||||
@@ -5,10 +5,9 @@ import aiohttp
|
||||
import aiofiles
|
||||
import os
|
||||
|
||||
from ndn_client import content_id,ndn_client
|
||||
from .pkg import pkg_info,pkg_media_info
|
||||
from .env import pkg_env
|
||||
|
||||
from ndn_client import ContentId,NDN_Client
|
||||
from .pkg import PackageInfo,PackageMediaInfo
|
||||
from .env import PackageEnv
|
||||
|
||||
logger = logging.getLogger(__name__)
|
||||
|
||||
@@ -19,8 +18,8 @@ INSTALL_TASK_STATE_DOWNLOADING = 3
|
||||
INSTALL_TASK_STATE_INSTALLING = 4
|
||||
INSTALL_TAKS_STATE_ERROR = 5
|
||||
|
||||
class pkg_install_task:
|
||||
def __init__(self,owner:pkg_env) -> None:
|
||||
class PackageInstallTask:
|
||||
def __init__(self,owner:PackageEnv) -> None:
|
||||
self.owner = owner
|
||||
self.state = INSTALL_TASK_STATE_CHECK_DEPENDENCY
|
||||
|
||||
@@ -29,20 +28,20 @@ class pkg_install_task:
|
||||
self.dependency_tasks = None
|
||||
self.error_str = None
|
||||
|
||||
class pkg_installer:
|
||||
def __init__(self,owner_env:pkg_env) -> None:
|
||||
class PackageInstaller:
|
||||
def __init__(self,owner_env:PackageEnv) -> None:
|
||||
self.all_tasks = {}
|
||||
self.owner_env = owner_env
|
||||
|
||||
def install(self,pkg_name:str,
|
||||
install_from_dependency = False, can_upgrade = True,skip_depends = False,options = None)->Tuple[pkg_install_task,str]:
|
||||
install_from_dependency = False, can_upgrade = True,skip_depends = False,options = None)->Tuple[PackageInstallTask,str]:
|
||||
|
||||
the_pkg_info : pkg_info = None
|
||||
the_pkg_info : PackageInfo = None
|
||||
is_upgrade : bool = False
|
||||
need_backup : bool = False
|
||||
|
||||
pkg_id,version_str,cid = pkg_info.parse_pkg_name(pkg_name)
|
||||
media_info : pkg_media_info = self.owner_env.get_media_info(pkg_name) # must use index-db?
|
||||
pkg_id,version_str,cid = PackageInfo.parse_pkg_name(pkg_name)
|
||||
media_info : PackageMediaInfo = self.owner_env.get_media_info(pkg_name) # must use index-db?
|
||||
if media_info is not None:
|
||||
if cid is not None:
|
||||
if can_upgrade:
|
||||
@@ -74,7 +73,7 @@ class pkg_installer:
|
||||
return result_task,"already installing"
|
||||
|
||||
logger.info(f"start download&install {pkg_name},install_from_dependency={install_from_dependency},upgrade={is_upgrade},backup={need_backup},target_pkg_info={the_pkg_info}")
|
||||
result_task = pkg_install_task()
|
||||
result_task = PackageInstallTask(self.owner_env)
|
||||
self.all_tasks[the_pkg_info.cid] = result_task
|
||||
async def download_and_install_pkg()->int:
|
||||
# check dependency
|
||||
@@ -104,7 +103,7 @@ class pkg_installer:
|
||||
|
||||
urls = self.owner_env.get_pkg_urls(the_pkg_info)
|
||||
#download
|
||||
client = ndn_client() # set watch
|
||||
client = NDN_Client() # set watch
|
||||
download_result = await client.get_file(the_pkg_info.cid,urls,target_full_path,options)
|
||||
if download_result !=0:
|
||||
result_task.state = INSTALL_TAKS_STATE_ERROR
|
||||
@@ -140,16 +139,16 @@ class pkg_installer:
|
||||
def uninstall(self):
|
||||
pass
|
||||
|
||||
def get_dependency_tasks(self,pkg:pkg_info,dependency_tasks):
|
||||
def get_dependency_tasks(self,pkg:PackageInfo,dependency_tasks):
|
||||
pass
|
||||
|
||||
async def check_dependency(self,pkg:pkg_info,task_list:{}) -> bool:
|
||||
async def check_dependency(self,pkg:PackageInfo,task_list:{}) -> bool:
|
||||
for depend_pkg_name in pkg.depends:
|
||||
depend_task = task_list.get(depend_pkg_name)
|
||||
if depend_task is not None:
|
||||
logger.debug(f"{pkg.name}'s depend pkg {depend_pkg_name} already in task list")
|
||||
continue
|
||||
depend_task = pkg_install_task()
|
||||
depend_task = PackageInstallTask(self.owner_env)
|
||||
task_list[depend_pkg_name] = depend_task
|
||||
|
||||
depend_pkg_info = self.owner_env.lookup(depend_pkg_name)
|
||||
|
||||
@@ -1,5 +1,7 @@
|
||||
from typing import Tuple
|
||||
|
||||
class pkg_info:
|
||||
|
||||
class PackageInfo:
|
||||
def __init__(self) -> None:
|
||||
self.name = ""
|
||||
self.cid = None
|
||||
@@ -20,7 +22,7 @@ class pkg_info:
|
||||
def cid(self) -> str:
|
||||
return self.cid
|
||||
|
||||
class pkg_media_info:
|
||||
class PackageMediaInfo:
|
||||
def __init__(self) -> None:
|
||||
pass
|
||||
|
||||
|
||||
Reference in New Issue
Block a user