mirror of
https://github.com/clearml/clearml-agent
synced 2025-01-31 17:16:51 +00:00
86 lines
3.8 KiB
Python
86 lines
3.8 KiB
Python
from collections import OrderedDict
|
|
from typing import Text
|
|
|
|
from .base import PackageManager
|
|
from .requirements import SimpleSubstitution
|
|
from ..base import safe_furl as furl
|
|
|
|
|
|
class ExternalRequirements(SimpleSubstitution):
|
|
|
|
name = "external_link"
|
|
|
|
def __init__(self, *args, **kwargs):
|
|
super(ExternalRequirements, self).__init__(*args, **kwargs)
|
|
self.post_install_req = []
|
|
self.post_install_req_lookup = OrderedDict()
|
|
|
|
def match(self, req):
|
|
# match both editable or code or unparsed
|
|
if not (not req.name or req.req and (req.req.editable or req.req.vcs)):
|
|
return False
|
|
if not req.req or not req.req.line or not req.req.line.strip() or req.req.line.strip().startswith('#'):
|
|
return False
|
|
return True
|
|
|
|
def post_install(self, session):
|
|
post_install_req = self.post_install_req
|
|
self.post_install_req = []
|
|
for req in post_install_req:
|
|
try:
|
|
freeze_base = PackageManager.out_of_scope_freeze() or ''
|
|
except:
|
|
freeze_base = ''
|
|
|
|
req_line = req.tostr(markers=False)
|
|
if req.req.vcs and req_line.startswith('git+'):
|
|
try:
|
|
url_no_frag = furl(req_line)
|
|
url_no_frag.set(fragment=None)
|
|
# reverse replace
|
|
fragment = req_line[::-1].replace(url_no_frag.url[::-1], '', 1)[::-1]
|
|
vcs_url = req_line[4:]
|
|
# reverse replace
|
|
vcs_url = vcs_url[::-1].replace(fragment[::-1], '', 1)[::-1]
|
|
from ..repo import Git
|
|
vcs = Git(session=session, url=vcs_url, location=None, revision=None)
|
|
vcs._set_ssh_url()
|
|
new_req_line = 'git+{}{}'.format(vcs.url_with_auth, fragment)
|
|
if new_req_line != req_line:
|
|
url_pass = furl(new_req_line).password
|
|
print('Replacing original pip vcs \'{}\' with \'{}\''.format(
|
|
req_line, new_req_line.replace(url_pass, '****', 1) if url_pass else new_req_line))
|
|
req_line = new_req_line
|
|
except Exception:
|
|
print('WARNING: Failed parsing pip git install, using original line {}'.format(req_line))
|
|
|
|
PackageManager.out_of_scope_install_package(req_line, "--no-deps")
|
|
try:
|
|
freeze_post = PackageManager.out_of_scope_freeze() or ''
|
|
package_name = list(set(freeze_post['pip']) - set(freeze_base['pip']))
|
|
if package_name and package_name[0] not in self.post_install_req_lookup:
|
|
self.post_install_req_lookup[package_name[0]] = req.req.line
|
|
except:
|
|
pass
|
|
if not PackageManager.out_of_scope_install_package(req_line, "--ignore-installed"):
|
|
raise ValueError("Failed installing GIT/HTTPs package \'{}\'".format(req_line))
|
|
|
|
def replace(self, req):
|
|
"""
|
|
Replace a requirement
|
|
:raises: ValueError if version is pre-release
|
|
"""
|
|
# Store in post req install, and return nothing
|
|
self.post_install_req.append(req)
|
|
# mark skip package, we will install it in post install hook
|
|
return Text('')
|
|
|
|
def replace_back(self, list_of_requirements):
|
|
if 'pip' in list_of_requirements:
|
|
original_requirements = list_of_requirements['pip']
|
|
list_of_requirements['pip'] = [r for r in original_requirements
|
|
if r not in self.post_install_req_lookup]
|
|
list_of_requirements['pip'] += [self.post_install_req_lookup.get(r, '')
|
|
for r in self.post_install_req_lookup.keys() if r in original_requirements]
|
|
return list_of_requirements
|