mirror of
https://github.com/LmeSzinc/StarRailCopilot.git
synced 2024-11-24 09:33:34 +00:00
133 lines
4.2 KiB
Python
133 lines
4.2 KiB
Python
import os
|
|
import re
|
|
import typing as t
|
|
from dataclasses import dataclass
|
|
from urllib.parse import urlparse
|
|
|
|
from deploy.Windows.config import DeployConfig
|
|
from deploy.Windows.logger import logger, Progress
|
|
from deploy.Windows.utils import cached_property
|
|
|
|
|
|
@dataclass
|
|
class DataDependency:
|
|
name: str
|
|
version: str
|
|
|
|
def __post_init__(self):
|
|
# uvicorn[standard] -> uvicorn
|
|
self.name = re.sub(r'\[.*\]', '', self.name)
|
|
# opencv_python -> opencv-python
|
|
self.name = self.name.replace('_', '-').strip()
|
|
# PyYaml -> pyyaml
|
|
self.name = self.name.lower()
|
|
self.version = self.version.strip()
|
|
self.version = re.sub(r'\.0$', '', self.version)
|
|
|
|
@cached_property
|
|
def pretty_name(self):
|
|
return f'{self.name}=={self.version}'
|
|
|
|
def __str__(self):
|
|
return self.pretty_name
|
|
|
|
__repr__ = __str__
|
|
|
|
def __eq__(self, other):
|
|
return str(self) == str(other)
|
|
|
|
def __hash__(self):
|
|
return hash(str(self))
|
|
|
|
|
|
class PipManager(DeployConfig):
|
|
@cached_property
|
|
def pip(self):
|
|
return f'"{self.python}" -m pip'
|
|
|
|
@cached_property
|
|
def python_site_packages(self):
|
|
return os.path.abspath(os.path.join(self.python, '../Lib/site-packages')) \
|
|
.replace(r"\\", "/").replace("\\", "/")
|
|
|
|
@cached_property
|
|
def set_installed_dependency(self) -> t.Set[DataDependency]:
|
|
data = []
|
|
regex = re.compile(r'(.*)-(.*).dist-info')
|
|
try:
|
|
for name in os.listdir(self.python_site_packages):
|
|
res = regex.search(name)
|
|
if res:
|
|
dep = DataDependency(name=res.group(1), version=res.group(2))
|
|
data.append(dep)
|
|
except FileNotFoundError:
|
|
logger.info(f'Directory not found: {self.python_site_packages}')
|
|
return set(data)
|
|
|
|
@cached_property
|
|
def set_required_dependency(self) -> t.Set[DataDependency]:
|
|
data = []
|
|
regex = re.compile('(.*)==(.*)[ ]*#')
|
|
file = self.filepath('./requirements.txt')
|
|
try:
|
|
with open(file, 'r', encoding='utf-8') as f:
|
|
for line in f.readlines():
|
|
res = regex.search(line)
|
|
if res:
|
|
dep = DataDependency(name=res.group(1), version=res.group(2))
|
|
data.append(dep)
|
|
except FileNotFoundError:
|
|
logger.info(f'File not found: {file}')
|
|
return set(data)
|
|
|
|
@cached_property
|
|
def set_dependency_to_install(self) -> t.Set[DataDependency]:
|
|
"""
|
|
A poor dependency comparison, but much much faster than `pip install` and `pip list`
|
|
"""
|
|
data = []
|
|
for dep in self.set_required_dependency:
|
|
if dep not in self.set_installed_dependency:
|
|
data.append(dep)
|
|
return set(data)
|
|
|
|
def pip_install(self):
|
|
logger.hr('Update Dependencies', 0)
|
|
|
|
if not self.InstallDependencies:
|
|
logger.info('InstallDependencies is disabled, skip')
|
|
Progress.UpdateDependency()
|
|
return
|
|
|
|
if not len(self.set_dependency_to_install):
|
|
logger.info('All dependencies installed')
|
|
Progress.UpdateDependency()
|
|
return
|
|
else:
|
|
logger.info(f'Dependencies to install: {self.set_dependency_to_install}')
|
|
|
|
# Install
|
|
logger.hr('Check Python', 1)
|
|
self.execute(f'"{self.python}" --version')
|
|
|
|
arg = []
|
|
if self.PypiMirror:
|
|
mirror = self.PypiMirror
|
|
arg += ['-i', mirror]
|
|
# Trust http mirror or skip ssl verify
|
|
if 'http:' in mirror or not self.SSLVerify:
|
|
arg += ['--trusted-host', urlparse(mirror).hostname]
|
|
elif not self.SSLVerify:
|
|
arg += ['--trusted-host', 'pypi.org']
|
|
arg += ['--trusted-host', 'files.pythonhosted.org']
|
|
|
|
# Don't update pip, just leave it.
|
|
# logger.hr('Update pip', 1)
|
|
# self.execute(f'"{self.pip}" install --upgrade pip{arg}')
|
|
arg += ['--disable-pip-version-check']
|
|
|
|
logger.hr('Update Dependencies', 1)
|
|
arg = ' ' + ' '.join(arg) if arg else ''
|
|
self.execute(f'{self.pip} install -r {self.requirements_file}{arg}')
|
|
Progress.UpdateDependency()
|