コード例 #1
0
    def execute(self, argv):
        if not argv:
            print("ERROR: Please specify a module name.\n")
            self.help()
            return False
        extra = {"command": "install"}

        if argv[0].endswith("tar.gz") or argv[0].endswith("phm"):
            result, tips, module_dir = default_module_manager.install_module(
                module_package=argv[0], extra=extra)
        elif os.path.exists(argv[0]) and os.path.isdir(argv[0]):
            result, tips, module_dir = default_module_manager.install_module(
                module_dir=argv[0], extra=extra)
        else:
            module_name = argv[0]
            module_version = None if "==" not in module_name else module_name.split(
                "==")[1]
            module_name = module_name if "==" not in module_name else module_name.split(
                "==")[0]
            CacheUpdater("hub_install", module_name, module_version).start()
            result, tips, module_dir = default_module_manager.install_module(
                module_name=module_name,
                module_version=module_version,
                extra=extra)

        print(tips)

        return True
コード例 #2
0
 def _init_with_name(self, name, version=None):
     log_msg = "Installing %s module" % name
     if version:
         log_msg += "-%s" % version
     logger.info(log_msg)
     result, tips, module_dir = default_module_manager.install_module(
         module_name=name, module_version=version)
     if not result:
         logger.error(tips)
         exit(1)
     logger.info(tips)
     self._init_with_module_file(module_dir[0])
コード例 #3
0
ファイル: run.py プロジェクト: zhongyunuestc/PaddleHub
    def find_module(self, module_name):
        module_dir = default_module_manager.search_module(module_name)
        if not module_dir:
            if os.path.exists(module_name):
                module_dir = (module_name, None)
            else:
                print("Install Module %s" % module_name)
                result, tips, module_dir = default_module_manager.install_module(
                    module_name)
                print(tips)
                if not result:
                    return None

        return hub.Module(module_dir=module_dir)
コード例 #4
0
ファイル: module.py プロジェクト: haozhe-an/PaddleHub
 def _init_with_name(self, name, version=None):
     log_msg = "Installing %s module" % name
     if version:
         log_msg += "-%s" % version
     logger.info(log_msg)
     extra = {"command": "install"}
     result, tips, module_dir = default_module_manager.install_module(
         module_name=name, module_version=version, extra=extra)
     if not result:
         logger.error(tips)
         raise RuntimeError(tips)
     else:
         logger.info(tips)
         self._init_with_module_file(module_dir[0])
コード例 #5
0
 def execute(self, argv):
     if not argv:
         print("ERROR: Please specify a module name.\n")
         self.help()
         return False
     module_name = argv[0]
     module_version = None if "==" not in module_name else module_name.split(
         "==")[1]
     module_name = module_name if "==" not in module_name else module_name.split(
         "==")[0]
     result, tips, module_dir = default_module_manager.install_module(
         module_name=module_name, module_version=module_version)
     print(tips)
     return True
コード例 #6
0
    def find_module(self, module_name):
        module_dir = default_module_manager.search_module(module_name)
        if not module_dir:
            if os.path.exists(module_name):
                module_dir = (module_name, None)
            else:
                print("Install Module %s" % module_name)
                extra = {"command": "install"}
                result, tips, module_dir = default_module_manager.install_module(
                    module_name, extra=extra)
                print(tips)
                if not result:
                    return None

        return hub.Module(directory=module_dir[0])
コード例 #7
0
ファイル: module.py プロジェクト: wxm2020/PaddleHub
    def init_with_name(cls, name, version=None, **kwargs):
        fp_lock = open(os.path.join(CACHE_HOME, name), "a")
        lock.flock(fp_lock, lock.LOCK_EX)
        log_msg = "Installing %s module" % name
        if version:
            log_msg += "-%s" % version
        logger.info(log_msg)
        extra = {"command": "install"}
        result, tips, module_dir = default_module_manager.install_module(
            module_name=name, module_version=version, extra=extra)
        if not result:
            logger.error(tips)
            raise RuntimeError(tips)

        logger.info(tips)
        lock.flock(fp_lock, lock.LOCK_UN)
        return cls.init_with_directory(directory=module_dir[0], **kwargs)