| import logging |
| |
| from ming import Session |
| from ming.orm.base import state |
| from ming.orm.ormsession import ThreadLocalORMSession, SessionExtension |
| |
| log = logging.getLogger(__name__) |
| |
| class ArtifactSessionExtension(SessionExtension): |
| |
| def __init__(self, session): |
| SessionExtension.__init__(self, session) |
| self.objects_added = [] |
| self.objects_modified = [] |
| self.objects_deleted = [] |
| |
| def before_flush(self, obj=None): |
| if obj is None: |
| self.objects_added = list(self.session.uow.new) |
| self.objects_modified = list(self.session.uow.dirty) |
| self.objects_deleted = list(self.session.uow.deleted) |
| else: # pragma no cover |
| st = state(obj) |
| if st.status == st.new: |
| self.objects_added = [ obj ] |
| elif st.status == st.dirty: |
| self.objects_modified = [ obj ] |
| elif st.status == st.deleted: |
| self.objects_deleted = [ obj ] |
| |
| def after_flush(self, obj=None): |
| "Update artifact references, and add/update this artifact to solr" |
| import allura.tasks.index_tasks |
| if not getattr(self.session, 'disable_artifact_index', False): |
| from pylons import g |
| from .index import ArtifactReference, Shortlink |
| from .session import main_orm_session |
| # Ensure artifact references & shortlinks exist for new objects |
| arefs = [] |
| try: |
| arefs = [ |
| ArtifactReference.from_artifact(obj) |
| for obj in self.objects_added + self.objects_modified ] |
| for obj in self.objects_added + self.objects_modified: |
| Shortlink.from_artifact(obj) |
| # Flush shortlinks |
| main_orm_session.flush() |
| except Exception: |
| log.exception("Failed to update artifact references. Is this a borked project migration?") |
| # Post delete and add indexing operations |
| if self.objects_deleted: |
| allura.tasks.index_tasks.del_artifacts.post( |
| [ obj.index_id() for obj in self.objects_deleted ]) |
| if arefs: |
| allura.tasks.index_tasks.add_artifacts.post([ aref._id for aref in arefs ]) |
| for obj in self.objects_added: |
| g.zarkov_event('create', extra=obj.index_id()) |
| for obj in self.objects_modified: |
| g.zarkov_event('modify', extra=obj.index_id()) |
| for obj in self.objects_deleted: |
| g.zarkov_event('delete', extra=obj.index_id()) |
| self.objects_added = [] |
| self.objects_modified = [] |
| self.objects_deleted = [] |
| |
| main_doc_session = Session.by_name('main') |
| project_doc_session = Session.by_name('project') |
| main_orm_session = ThreadLocalORMSession(main_doc_session) |
| project_orm_session = ThreadLocalORMSession(project_doc_session) |
| artifact_orm_session = ThreadLocalORMSession( |
| doc_session=project_doc_session, |
| extensions = [ ArtifactSessionExtension ]) |
| repository_orm_session = ThreadLocalORMSession( |
| doc_session=main_doc_session, |
| extensions = [ ]) |