Initial changes to support the new Django 1.6 project structure
This commit is contained in:
Roberto Rosario
176
mayan/apps/installation/models.py
Normal file
176
mayan/apps/installation/models.py
Normal file
@@ -0,0 +1,176 @@
|
||||
from __future__ import absolute_import
|
||||
|
||||
import os
|
||||
import sys
|
||||
import platform
|
||||
import uuid
|
||||
import time
|
||||
|
||||
import pbs
|
||||
import psutil
|
||||
import requests
|
||||
from git import Repo
|
||||
|
||||
try:
|
||||
from pbs import lsb_release, uname
|
||||
except pbs.CommandNotFound:
|
||||
LSB = False
|
||||
else:
|
||||
LSB = True
|
||||
|
||||
|
||||
from django.db import models
|
||||
from django.utils.translation import ugettext_lazy as _
|
||||
from django.utils.datastructures import SortedDict
|
||||
from django.conf import settings
|
||||
|
||||
from common.models import Singleton
|
||||
from common.utils import pretty_size
|
||||
from main import __version__ as mayan_version
|
||||
from lock_manager import Lock, LockError
|
||||
from ocr.conf.settings import TESSERACT_PATH, UNPAPER_PATH, PDFTOTEXT_PATH
|
||||
|
||||
from .classes import Property, PropertyNamespace, VirtualEnv, PIPNotFound
|
||||
|
||||
FORM_SUBMIT_URL = 'https://docs.google.com/spreadsheet/formResponse'
|
||||
FORM_KEY = 'dGZrYkw3SDl5OENMTG15emp1UFFEUWc6MQ'
|
||||
FORM_RECEIVER_FIELD = 'entry.0.single'
|
||||
TIMEOUT = 5
|
||||
FABFILE_MARKER = os.path.join(settings.BASE_DIR, 'fabfile_install')
|
||||
|
||||
|
||||
class Installation(Singleton):
|
||||
_properties = SortedDict()
|
||||
|
||||
is_first_run = models.BooleanField(default=False)
|
||||
uuid = models.CharField(max_length=48, blank=True, default=lambda: unicode(uuid.uuid4()))
|
||||
|
||||
def add_property(self, property_instance):
|
||||
self._properties[property_instance.name] = property_instance
|
||||
|
||||
def get_properties(self):
|
||||
self.set_properties()
|
||||
return self._properties.values()
|
||||
|
||||
def os_properties(self):
|
||||
namespace = PropertyNamespace('os', _(u'Operating system'))
|
||||
if LSB:
|
||||
namespace.add_property('is_lsb', _(u'LSB OS'), True, True)
|
||||
namespace.add_property('distributor_id', _(u'Distributor ID'), lsb_release('-i', '-s'), True)
|
||||
namespace.add_property('description', _(u'Description'), lsb_release('-d', '-s'), True)
|
||||
namespace.add_property('release', _(u'Release'), lsb_release('-r', '-s'), True)
|
||||
namespace.add_property('codename', _(u'Codename'), lsb_release('-c', '-s'), True)
|
||||
namespace.add_property('sysinfo', _(u'System info'), uname('-a'), True)
|
||||
else:
|
||||
namespace.add_property('is_lsb', _(u'LSB OS'), False)
|
||||
|
||||
namespace.add_property('architecture', _(u'OS architecture'), platform.architecture(), report=True)
|
||||
namespace.add_property('python_version', _(u'Python version'), platform.python_version(), report=True)
|
||||
namespace.add_property('hostname', _(u'Hostname'), platform.node())
|
||||
namespace.add_property('platform', _(u'Platform'), sys.platform, report=True)
|
||||
namespace.add_property('machine', _(u'Machine'), platform.machine(), report=True)
|
||||
namespace.add_property('processor', _(u'Processor'), platform.processor(), report=True)
|
||||
namespace.add_property('cpus', _(u'Number of CPUs'), psutil.NUM_CPUS, report=True)
|
||||
namespace.add_property('total_phymem', _(u'Total physical memory'), pretty_size(psutil.TOTAL_PHYMEM), report=True)
|
||||
namespace.add_property('disk_partitions', _(u'Disk partitions'), '; '.join(['%s %s %s %s' % (partition.device, partition.mountpoint, partition.fstype, partition.opts) for partition in psutil.disk_partitions()]))
|
||||
|
||||
def binary_dependencies(self):
|
||||
namespace = PropertyNamespace('bins', _(u'Binary dependencies'))
|
||||
|
||||
tesseract = pbs.Command(TESSERACT_PATH)
|
||||
try:
|
||||
namespace.add_property('tesseract', _(u'tesseract version'), tesseract('-v').stderr, report=True)
|
||||
except pbs.CommandNotFound:
|
||||
namespace.add_property('tesseract', _(u'tesseract version'), _(u'not found'), report=True)
|
||||
except Exception:
|
||||
namespace.add_property('tesseract', _(u'tesseract version'), _(u'error getting version'), report=True)
|
||||
|
||||
unpaper = pbs.Command(UNPAPER_PATH)
|
||||
try:
|
||||
namespace.add_property('unpaper', _(u'unpaper version'), unpaper('-V').stdout, report=True)
|
||||
except pbs.CommandNotFound:
|
||||
namespace.add_property('unpaper', _(u'unpaper version'), _(u'not found'), report=True)
|
||||
except Exception:
|
||||
namespace.add_property('unpaper', _(u'unpaper version'), _(u'error getting version'), report=True)
|
||||
|
||||
pdftotext = pbs.Command(PDFTOTEXT_PATH)
|
||||
try:
|
||||
namespace.add_property('pdftotext', _(u'pdftotext version'), pdftotext('-v').stderr, report=True)
|
||||
except pbs.CommandNotFound:
|
||||
namespace.add_property('pdftotext', _(u'pdftotext version'), _(u'not found'), report=True)
|
||||
except Exception:
|
||||
namespace.add_property('pdftotext', _(u'pdftotext version'), _(u'error getting version'), report=True)
|
||||
|
||||
def mayan_properties(self):
|
||||
namespace = PropertyNamespace('mayan', _(u'Mayan EDMS'))
|
||||
|
||||
namespace.add_property('uuid', _(u'UUID'), self.uuid, report=True)
|
||||
namespace.add_property('mayan_version', _(u'Mayan EDMS version'), mayan_version, report=True)
|
||||
namespace.add_property('fabfile', _(u'Installed via fabfile'), os.path.exists(FABFILE_MARKER), report=True)
|
||||
|
||||
def git_properties(self):
|
||||
namespace = PropertyNamespace('git', _(u'Git repository'))
|
||||
|
||||
try:
|
||||
repo = Repo(os.path.abspath(os.path.join(settings.BASE_DIR, '..')))
|
||||
except:
|
||||
namespace.add_property(Property('is_git_repo', _(u'Running from a Git repository'), False))
|
||||
else:
|
||||
repo.config_reader()
|
||||
headcommit = repo.head.commit
|
||||
namespace.add_property('is_git_repo', _(u'Running from a Git repository'), True)
|
||||
namespace.add_property('repo_remotes', _(u'Repository remotes'), ', '.join([unicode(remote) for remote in repo.remotes]), report=True)
|
||||
namespace.add_property('repo_remotes_urls', _(u'Repository remotes URLs'), ', '.join([unicode(remote.url) for remote in repo.remotes]), report=True)
|
||||
namespace.add_property('repo_head_reference', _(u'Branch'), repo.head.reference, report=True)
|
||||
namespace.add_property('headcommit_hexsha', _(u'HEAD commit hex SHA'), headcommit.hexsha, report=True)
|
||||
namespace.add_property('headcommit_author', _(u'HEAD commit author'), headcommit.author)
|
||||
namespace.add_property('headcommit_authored_date', _(u'HEAD commit authored date'), time.asctime(time.gmtime(headcommit.authored_date)), report=True)
|
||||
namespace.add_property('headcommit_committer', _(u'HEAD commit committer'), headcommit.committer)
|
||||
namespace.add_property('headcommit_committed_date', _(u'HEAD commit committed date'), time.asctime(time.gmtime(headcommit.committed_date)), report=True)
|
||||
namespace.add_property('headcommit_message', _(u'HEAD commit message'), headcommit.message, report=True)
|
||||
|
||||
def virtualenv_properties(self):
|
||||
namespace = PropertyNamespace('venv', _(u'VirtuanEnv'))
|
||||
try:
|
||||
venv = VirtualEnv()
|
||||
except PIPNotFound:
|
||||
namespace.add_property('pip', 'pip', _(u'pip not found.'), report=True)
|
||||
else:
|
||||
for item, version, result in venv.get_results():
|
||||
namespace.add_property(item, '%s (%s)' % (item, version), result, report=True)
|
||||
|
||||
def set_properties(self):
|
||||
self._properties = SortedDict()
|
||||
self.os_properties()
|
||||
self.binary_dependencies()
|
||||
self.mayan_properties()
|
||||
self.git_properties()
|
||||
self.virtualenv_properties()
|
||||
|
||||
def __getattr__(self, name):
|
||||
self.set_properties()
|
||||
try:
|
||||
return self._properties[name].value
|
||||
except KeyError:
|
||||
raise AttributeError
|
||||
|
||||
def submit(self):
|
||||
try:
|
||||
lock = Lock.acquire_lock('upload_stats')
|
||||
except LockError:
|
||||
pass
|
||||
else:
|
||||
self.set_properties()
|
||||
|
||||
try:
|
||||
requests.post(FORM_SUBMIT_URL, data={'formkey': FORM_KEY, FORM_RECEIVER_FIELD: Property.get_reportable(as_json=True)}, timeout=TIMEOUT)
|
||||
except (requests.exceptions.Timeout, requests.exceptions.ConnectionError):
|
||||
pass
|
||||
else:
|
||||
self.is_first_run = False
|
||||
self.save()
|
||||
finally:
|
||||
lock.release()
|
||||
|
||||
class Meta:
|
||||
verbose_name = verbose_name_plural = _(u'installation details')
|
||||
Reference in New Issue
Block a user