Merge branch 'master' of https://github.com/iksaif/euscan
This commit is contained in:
		@@ -1,108 +0,0 @@
 | 
				
			|||||||
"""
 | 
					 | 
				
			||||||
djeuscan.helpers
 | 
					 | 
				
			||||||
"""
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
from distutils.version import StrictVersion, LooseVersion
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
def xint(i):
 | 
					 | 
				
			||||||
    """
 | 
					 | 
				
			||||||
    Tries to cast to int, fallbacks to 0
 | 
					 | 
				
			||||||
    """
 | 
					 | 
				
			||||||
    try:
 | 
					 | 
				
			||||||
        return int(i)
 | 
					 | 
				
			||||||
    except Exception:
 | 
					 | 
				
			||||||
        return 0
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
def select_related_last_versions(queryset):
 | 
					 | 
				
			||||||
    queryset = queryset.select_related(
 | 
					 | 
				
			||||||
        'last_version_gentoo',
 | 
					 | 
				
			||||||
        'last_version_overlay',
 | 
					 | 
				
			||||||
        'last_version_upstream'
 | 
					 | 
				
			||||||
    )
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
def version_key(version):
 | 
					 | 
				
			||||||
    version = version.version
 | 
					 | 
				
			||||||
    try:
 | 
					 | 
				
			||||||
        return StrictVersion(version)
 | 
					 | 
				
			||||||
    # in case of abnormal version number, fall back to LooseVersion
 | 
					 | 
				
			||||||
    except ValueError:
 | 
					 | 
				
			||||||
        return LooseVersion(version)
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
def packages_from_names(data):
 | 
					 | 
				
			||||||
    """
 | 
					 | 
				
			||||||
    Returns a list of Package objects from a string of names
 | 
					 | 
				
			||||||
    """
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
    from djeuscan.models import Package
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
    packages = []
 | 
					 | 
				
			||||||
    data = data.replace("\r", "")
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
    for pkg in data.split('\n'):
 | 
					 | 
				
			||||||
        if '/' in pkg:
 | 
					 | 
				
			||||||
            cat, pkg = pkg.split('/')
 | 
					 | 
				
			||||||
            packages.extend(Package.objects.filter(category=cat, name=pkg))
 | 
					 | 
				
			||||||
        else:
 | 
					 | 
				
			||||||
            packages.extend(Package.objects.filter(name=pkg))
 | 
					 | 
				
			||||||
    return packages
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
def rename_fields(vqs, fields):
 | 
					 | 
				
			||||||
    ret = []
 | 
					 | 
				
			||||||
    for n in vqs:
 | 
					 | 
				
			||||||
        for tr in fields:
 | 
					 | 
				
			||||||
            if tr[0] in n:
 | 
					 | 
				
			||||||
                n[tr[1]] = n[tr[0]]
 | 
					 | 
				
			||||||
                del n[tr[0]]
 | 
					 | 
				
			||||||
        ret.append(n)
 | 
					 | 
				
			||||||
    return ret
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
class catch_and_return(object):
 | 
					 | 
				
			||||||
    def __init__(self, err, response):
 | 
					 | 
				
			||||||
        self.err = err
 | 
					 | 
				
			||||||
        self.response = response
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
    def __call__(self, fn):
 | 
					 | 
				
			||||||
        def wrapper(*args, **kwargs):
 | 
					 | 
				
			||||||
            try:
 | 
					 | 
				
			||||||
                return fn(*args, **kwargs)
 | 
					 | 
				
			||||||
            except self.err:
 | 
					 | 
				
			||||||
                return self.response
 | 
					 | 
				
			||||||
        return wrapper
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
def get_account_categories(user):
 | 
					 | 
				
			||||||
    from djeuscan.models import Package, CategoryAssociation
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
    # TODO: This is quite ugly
 | 
					 | 
				
			||||||
    category_names = [obj.category for obj in
 | 
					 | 
				
			||||||
                      CategoryAssociation.objects.filter(user=user)]
 | 
					 | 
				
			||||||
    return [c for c in Package.objects.categories()
 | 
					 | 
				
			||||||
            if c["category"] in category_names]
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
def get_account_herds(user):
 | 
					 | 
				
			||||||
    from djeuscan.models import Package, HerdAssociation
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
    ids = [obj.herd.pk for obj in
 | 
					 | 
				
			||||||
           HerdAssociation.objects.filter(user=user)]
 | 
					 | 
				
			||||||
    return Package.objects.herds(ids=ids)
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
def get_account_maintainers(user):
 | 
					 | 
				
			||||||
    from djeuscan.models import Package, MaintainerAssociation
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
    ids = [obj.maintainer.pk for obj in
 | 
					 | 
				
			||||||
           MaintainerAssociation.objects.filter(user=user)]
 | 
					 | 
				
			||||||
    return Package.objects.maintainers(ids=ids)
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
def get_account_packages(user):
 | 
					 | 
				
			||||||
    from djeuscan.models import PackageAssociation
 | 
					 | 
				
			||||||
    return [obj.package for obj in
 | 
					 | 
				
			||||||
            PackageAssociation.objects.filter(user=user)]
 | 
					 | 
				
			||||||
@@ -13,9 +13,6 @@ from euscan.version import get_version_type
 | 
				
			|||||||
from djeuscan.processing import FakeLogger
 | 
					from djeuscan.processing import FakeLogger
 | 
				
			||||||
from djeuscan.models import Package, Version, VersionLog
 | 
					from djeuscan.models import Package, Version, VersionLog
 | 
				
			||||||
 | 
					
 | 
				
			||||||
PORTDB = None
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
class ScanPortage(object):
 | 
					class ScanPortage(object):
 | 
				
			||||||
    def __init__(self, logger=None, no_log=False, purge_packages=False,
 | 
					    def __init__(self, logger=None, no_log=False, purge_packages=False,
 | 
				
			||||||
                 purge_versions=False):
 | 
					                 purge_versions=False):
 | 
				
			||||||
@@ -24,10 +21,6 @@ class ScanPortage(object):
 | 
				
			|||||||
        self.purge_packages = purge_packages
 | 
					        self.purge_packages = purge_packages
 | 
				
			||||||
        self.purge_versions = purge_versions
 | 
					        self.purge_versions = purge_versions
 | 
				
			||||||
 | 
					
 | 
				
			||||||
        global PORTDB
 | 
					 | 
				
			||||||
        if not PORTDB:  # Lazy loading for portdb
 | 
					 | 
				
			||||||
            PORTDB = portage.db[portage.root]["porttree"].dbapi
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
        self.style = color_style()
 | 
					        self.style = color_style()
 | 
				
			||||||
 | 
					
 | 
				
			||||||
        self._cache = {'packages': {}, 'versions': {}}
 | 
					        self._cache = {'packages': {}, 'versions': {}}
 | 
				
			||||||
@@ -231,7 +224,7 @@ class ScanPortage(object):
 | 
				
			|||||||
            package.category, package.name, ver, rev, slot, overlay
 | 
					            package.category, package.name, ver, rev, slot, overlay
 | 
				
			||||||
        )
 | 
					        )
 | 
				
			||||||
 | 
					
 | 
				
			||||||
        overlay_path = overlay_path or PORTDB.settings["PORTDIR"]
 | 
					        overlay_path = overlay_path or portage.settings["PORTDIR"]
 | 
				
			||||||
        package_path = join(overlay_path, package.category, package.name)
 | 
					        package_path = join(overlay_path, package.category, package.name)
 | 
				
			||||||
        ebuild_path = join(package_path, "%s.ebuild" % cpv.split("/")[-1])
 | 
					        ebuild_path = join(package_path, "%s.ebuild" % cpv.split("/")[-1])
 | 
				
			||||||
        metadata_path = join(package_path, "metadata.xml")
 | 
					        metadata_path = join(package_path, "metadata.xml")
 | 
				
			||||||
 
 | 
				
			|||||||
		Reference in New Issue
	
	Block a user