167 lines
4.6 KiB
Python
167 lines
4.6 KiB
Python
from __future__ import absolute_import
|
|
|
|
from collections import namedtuple
|
|
import os
|
|
import sys
|
|
|
|
import pbs
|
|
|
|
try:
|
|
from pbs import pip
|
|
PIP = True
|
|
except pbs.CommandNotFound:
|
|
PIP = False
|
|
|
|
from django.conf import settings
|
|
from django.utils.simplejson import dumps
|
|
|
|
|
|
class PIPNotFound(Exception):
|
|
pass
|
|
|
|
|
|
class PropertyNamespace(object):
|
|
_registry = {}
|
|
|
|
@classmethod
|
|
def get(cls, name):
|
|
return cls._registry[name]
|
|
|
|
@classmethod
|
|
def get_all(cls):
|
|
return cls._registry.values()
|
|
|
|
def __init__(self, name, label):
|
|
self.name = name
|
|
self.label = label
|
|
self.properties = {}
|
|
self.__class__._registry[name] = self
|
|
|
|
def __unicode__(self):
|
|
return unicode(self.label)
|
|
|
|
def __str__(self):
|
|
return str(self.label)
|
|
|
|
def add_property(self, *args, **kwargs):
|
|
prop = Property(*args, **kwargs)
|
|
self.properties[prop.name] = prop
|
|
|
|
def get_properties(self):
|
|
return self.properties.values()
|
|
|
|
@property
|
|
def id(self):
|
|
return self.name
|
|
|
|
|
|
class Property(object):
|
|
_registry = {}
|
|
|
|
@classmethod
|
|
def get_all(cls):
|
|
return cls._registry.values()
|
|
|
|
@classmethod
|
|
def get(cls, name):
|
|
return cls._registry[name]
|
|
|
|
@classmethod
|
|
def get_reportable(cls, as_dict=False, as_json=False):
|
|
if as_json:
|
|
return dumps(cls.get_reportable(as_dict=True))
|
|
|
|
if not as_dict:
|
|
return [prop for prop in cls.get_all() if prop.report]
|
|
else:
|
|
result = {}
|
|
for prop in cls.get_all():
|
|
if prop.report:
|
|
result[prop.name] = unicode(prop.value)
|
|
return result
|
|
|
|
def __init__(self, name, label, value, report=False):
|
|
self.name = name
|
|
self.label = label
|
|
self.value = value
|
|
self.report = report
|
|
self.__class__._registry[name] = self
|
|
|
|
def __unicode__(self):
|
|
return unicode(self.value)
|
|
|
|
def __str__(self):
|
|
return str(self.value)
|
|
|
|
|
|
Dependency = namedtuple('Dependency', 'name, version, standard')
|
|
|
|
|
|
class VirtualEnv(object):
|
|
def extract_dependency(self, string):
|
|
string = str(string.strip())
|
|
|
|
try:
|
|
package, version = string.split('==')
|
|
except ValueError:
|
|
# item is not installed from package, svn/git maybe
|
|
try:
|
|
version, package = string.split('=')
|
|
except:
|
|
# has no version number
|
|
return Dependency(string, version=None, standard=True)
|
|
else:
|
|
version = version.split('#')[0].split(' ')[1] # Get rid of '#egg' and '-e'
|
|
return Dependency(package, version, standard=False)
|
|
else:
|
|
return Dependency(package, version, standard=True)
|
|
|
|
|
|
def get_packages_info(self, requirements_file=None):
|
|
if requirements_file:
|
|
with open(requirements_file) as file_in:
|
|
for line in file_in.readlines():
|
|
yield self.extract_dependency(line)
|
|
else:
|
|
for item in pip('freeze').splitlines():
|
|
yield self.extract_dependency(item)
|
|
|
|
|
|
def __init__(self):
|
|
self.requirements_file_path = os.path.join(settings.PROJECT_ROOT, 'requirements', 'production.txt')
|
|
if not PIP:
|
|
raise PIPNotFound
|
|
|
|
|
|
def get_results(self):
|
|
requirements = {}
|
|
installed_packages = {}
|
|
|
|
for item in self.get_packages_info(self.requirements_file_path):
|
|
requirements[item.name] = item
|
|
|
|
for item in self.get_packages_info():
|
|
installed_packages[item.name] = item
|
|
|
|
for name, item in requirements.items():
|
|
try:
|
|
if item.standard:
|
|
if item.version:
|
|
if item.version == installed_packages[name].version:
|
|
status = item.version
|
|
else:
|
|
status = installed_packages[name].version
|
|
else:
|
|
status = None
|
|
else:
|
|
# Non standard version number, check SVN or GIT path
|
|
if item.version == installed_packages['%s-dev' % name.replace('-', '_')].version:
|
|
status = item.version
|
|
else:
|
|
status = installed_packages['%s-dev' % name.replace('-', '_')].version
|
|
except KeyError:
|
|
# Not installed package found matching with name matchin requirement
|
|
status = False
|
|
|
|
yield name, item.version, status
|