class MainSiteImagesView(BaseResource):
def __init__(self, request):
super(MainSiteImagesView, self).__init__(request)
settings = request.registry.settings
self.imagepath = settings['default.siteimages.directory']
self.mgr = SiteImageManager(request.db, self.imagepath)
def serialize_object(self, dbobj):
data = dbobj.serialize()
data['thumbnail'] = base64.b64encode(dbobj.thumbnail)
return data
def collection_query(self):
return self.mgr.query()
def collection_post(self):
data = self.request.POST
name = data['imagefile']
upload = data['imagefile']
name = upload.filename
fp = upload.file
try:
image = self.mgr.add_image(name, fp)
return dict(image=image.serialize(), result='success')
except FilenameInDatabaseError:
raise HTTPConflict("This file already exists.")
except ImageFileExistsError:
raise HTTPConflict("This file already exists.")
def __init__(self, session, dburl, annex_directory,
siteimages_directory):
super(MainDBAdminManager, self).__init__(session)
self.annex_directory = annex_directory
self.siteimages_directory = siteimages_directory
self.dburl = dburl
self.annex_repomgr = AnnexRepoManager(session, annex_directory)
self.annex_filemgr = AnnexFileManager(session, annex_directory)
self.siteimage_manager = SiteImageManager(session,
self.siteimages_directory)
class MainDBAdminManager(BaseManager):
def __init__(self, session, dburl, annex_directory,
siteimages_directory):
super(MainDBAdminManager, self).__init__(session)
self.annex_directory = annex_directory
self.siteimages_directory = siteimages_directory
self.dburl = dburl
self.annex_repomgr = AnnexRepoManager(session, annex_directory)
self.annex_filemgr = AnnexFileManager(session, annex_directory)
self.siteimage_manager = SiteImageManager(session,
self.siteimages_directory)
def _setup_repos(self):
dbmodel = self.annex_repomgr.dbmodel
if not self.session.query(dbmodel).count():
self.annex_repomgr.setup_repositories()
def get_annex_info(self, inspector=None, new_job=False):
if inspector is None:
inspector = celery_app.control.inspect()
result = inspector.app.AsyncResult(POPULATE_ANNEXDB_TASK)
status = result.status
populated = False
if status in ['SUCCESS', 'STARTED']:
populated = True
repos = self.annex_repomgr.query().count()
keys = self.annex_filemgr.keymgr.query().count()
files = self.annex_filemgr.query().count()
data = dict(status=status, repos=repos, keys=keys, files=files,
populated=populated, new_job=new_job)
return data
def get_siteimage_info(self):
images = self.siteimage_manager.query().count()
return dict(images=images)
return data
def delete_annex_db(self):
self.annex_filemgr.delete_everything_tm()
ip = celery_app.control.inspect()
result = ip.app.AsyncResult(POPULATE_ANNEXDB_TASK)
if result.status == 'SUCCESS':
result.forget()
return self.get_annex_info(inspector=ip)
def delete_siteimages(self):
self.siteimage_manager.delete_everything()
def populate_annexdb(self):
new_job = False
ip = celery_app.control.inspect()
result = ip.app.AsyncResult(POPULATE_ANNEXDB_TASK)
if result.status == 'PENDING':
self._setup_repos()
args = (self.dburl, self.annex_directory)
result = populate_annex_files.apply_async(
args, task_id=POPULATE_ANNEXDB_TASK,
countdown=0)
new_job = True
return self.get_annex_info(inspector=ip, new_job=new_job)
def get_all_info(self, inspector=None):
data = dict(gitannex=self.get_annex_info(inspector=inspector),
siteimages=self.get_siteimage_info())
return data
def __init__(self, request):
super(MainSiteImagesView, self).__init__(request)
settings = request.registry.settings
self.imagepath = settings['default.siteimages.directory']
self.mgr = SiteImageManager(request.db, self.imagepath)