guest_worker/sorteerhoed/HITStore.py

import logging
from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy import Column, Integer, String, DateTime, Float
from sqlalchemy.orm import relationship
from sqlalchemy.sql.schema import ForeignKey, Sequence
from sqlalchemy.engine import create_engine
from sqlalchemy.orm.session import sessionmaker
import datetime
from contextlib import contextmanager
import uuid
import os

mainLogger = logging.getLogger("sorteerhoed")
logger = mainLogger.getChild("store")

Base = declarative_base()

"""
HIT lifetime:


created
accepted
(returned!)
working
awaiting amazon confirmation (submitted on page)
submitted

Actions:
creating Hit (creating hit with scanned image)

Scanning

"""

class HIT(Base):
    __tablename__ = 'hits'
    id = Column(Integer, Sequence('hit_id'), primary_key=True) # our sequential hit id
    hit_id = Column(String(255)) # amazon's hit id
    created_at = Column(DateTime, default=datetime.datetime.utcnow)
    updated_at = Column(DateTime, default=datetime.datetime.utcnow)
    scanned_at = Column(DateTime, default=None)
    plotted_at = Column(DateTime, default=None)
    deleted_at = Column(DateTime, default=None)
    assignments = relationship("Assignment", back_populates="hit", order_by="Assignment.created_at")
    fee = Column(Float(precision=2), default=None)

    # previous hit so we can load the corrent image
    # previous_hit_id = Column(Integer, ForeignKey('hits.id'), default=None)
    # previous_hit = relationship("HIT")

    def getImagePath(self):
        return os.path.join('scanimation/interfaces/frames', f"{self.id:06d}.jpg")
    
    def getImageUrl(self):
        return os.path.join('/frames', f"{self.id:06d}.jpg")

    def getSvgImageUrl(self):
        return f"scans/{self.id:06d}.svg"

    def getSvgImagePath(self):
        return os.path.join('www', self.getSvgImageUrl())

    def getLastAssignment(self):
        if not len(self.assignments):
            return None
        return self.assignments[-1]

    def getAssignmentById(self, assignmentId):
        for a in self.assignments:
            if a.assignment_id == assignmentId:
                return a
        return None

    def getStatus(self):
        assignment = self.getLastAssignment()
        if self.deleted_at:
            return "deleted"
        if not self.hit_id:
            return "creating"
        if not assignment:
            return "awaiting worker"
        if self.scanned_at:
            return "scanned"
        return assignment.getStatus()
    
    def toDict(self) -> dict:
        values = {c.name: getattr(self, c.name) for c in self.__table__.columns}
        assignment = self.getLastAssignment()
        values['assignment'] = assignment.toDict() if assignment else None
        values['state'] = self.getStatus()
        values['scan_image'] = self.getImageUrl() if self.scanned_at else None
        values['svg_image'] = self.getSvgImageUrl() if self.isSubmitted() else None
        return values

    def delete(self):
        self.deleted_at = datetime.datetime.utcnow()
        
    def isSubmitted(self) -> bool:
        a = self.getLastAssignment()
        if not a:
            return False
        return bool(a.submit_page_at)
    
    def isConfirmed(self) -> bool:
        a = self.getLastAssignment()
        if not a:
            return False
        return bool(a.confirmed_at)

class Assignment(Base):
    __tablename__ = 'assignments'
    id = Column(Integer, Sequence('assignment_id'), primary_key=True) # our sequential hit id
    assignment_id = Column(String(255)) # amazon's assignment id
    hit_id = Column(Integer, ForeignKey('hits.id')) # our sequential hit id
    hit = relationship("HIT", back_populates="assignments")
    uuid = Column(String(32), default=lambda : uuid.uuid4().hex)

    created_at = Column(DateTime, default=datetime.datetime.utcnow)
    updated_at = Column(DateTime, default=datetime.datetime.utcnow)

    assignment_id = Column(String(255), default = None)
    worker_id = Column(String(255), default = None)
    accept_at = Column(DateTime, default=None)  # accept time acccording to SQS
    # open_page_at = Column(DateTime, default=None)
    submit_page_at = Column(DateTime, default=None) # Submit the page
    confirmed_at = Column(DateTime, default=None) # validate with UUID when getting Message from Amazon
    abandoned_at = Column(DateTime, default=None)
    rejected_at = Column(DateTime, default=None)

    answer = Column(String(255), default=None)
    turk_ip = Column(String(255), default=None)
    turk_country = Column(String(255), default=None)
    turk_os = Column(String(255), default=None)
    turk_browser = Column(String(255), default=None)
    
    def getStatus(self):
        if self.rejected_at:
            return "rejected"
        if self.abandoned_at:
            return "abandoned"
        if not self.submit_page_at:
            return "working"
        if not self.confirmed_at:
            return "submitted"
        return "confirmed"
    
    def toDict(self) -> dict:
        values = {c.name: getattr(self, c.name) for c in self.__table__.columns}
        return values


class Store:
    def __init__(self, db_filename, logLevel=0):
        path = os.path.abspath(db_filename)
        if logLevel <= logging.DEBUG:
            logging.getLogger('sqlalchemy.engine').setLevel(logging.INFO)

        self.engine = create_engine('sqlite:///'+path, echo=False, connect_args={'check_same_thread': False})
        Base.metadata.create_all(self.engine)
        self.Session = sessionmaker(bind=self.engine)
        self.session = self.Session()

        self.currentHit = None # mirrors Centralmanagmenet, stored here so we can quickly access it from webserver classes
        self.updateHooks = []
    
    def registerUpdateHook(self, hook):
        if hook not in self.updateHooks:
            logger.info(f"Register update hook: {hook}")
            self.updateHooks.append(hook)
    
    def triggerUpdateHooks(self, hit = None):
        for hook in self.updateHooks:
            if callable(hook):  # it's a method
                hook(hit)
            else:  # assume it's an object
                hook.update(hit)

    @contextmanager
    def getSession(self):
        """Provide a transactional scope around a series of operations."""
        try:
            yield self.session
            self.session.commit()
        except:
            self.session.rollback()
            raise

    def getHits(self):
        return self.session.query(HIT).order_by(HIT.created_at.desc())

    def getHitById(self, hitId):
        return self.session.query(HIT).\
          filter(HIT.id==hitId).one()

    def getHitByRemoteId(self, amazonHitId):
        return self.session.query(HIT).\
          filter(HIT.hit_id==amazonHitId).one()


    def getLastSubmittedHit(self):
        return self.session.query(HIT).\
            join(Assignment).\
            filter(Assignment.submit_page_at!=None).\
            order_by(HIT.created_at.desc()).first()
    
    def getNewestHits(self, n = 2) -> list:
        hits = list(
            self.session.query(HIT).\
              filter(HIT.deleted_at==None).\
              order_by(HIT.created_at.desc()).limit(2)
        ) 
        # select DESC, because we want latest, then reverse list to get in right order
        hits.reverse()
        return hits

    def createHIT(self) -> HIT:
        with self.getSession() as s:
            hit = HIT()
            s.add(hit)
            s.flush()
            s.refresh(hit)
            logger.info(f"Created HIT {hit.id}")
        
        self.triggerUpdateHooks(hit)
        
        return hit

    def newAssignment(self, hit: HIT, assignmentId) -> Assignment:
        # TODO: reset() central management if has pending lastAssignment()
        
        with self.getSession() as s:
            assignment = Assignment()
            assignment.assignment_id = assignmentId
            hit.assignments.append(assignment)
            s.add(assignment)
            s.flush()
            s.refresh(hit)
            logger.info(f"Created Assignment {assignment.id}")
        
        self.triggerUpdateHooks(hit)
        
        return assignment

    def saveHIT(self, hit):
        with self.getSession() as s:
            logger.info(f"Updating hit! {hit.id}")
#             s.flush()
            self.triggerUpdateHooks(hit)

    def saveAssignment(self, assignment):
        with self.getSession() as s:
            logger.info(f"Updating assignment! {assignment.id}")
#             s.flush()
            self.triggerUpdateHooks(assignment.hit)

#     def addHIT(self, hit: HIT):
#         with self.getSession() as s:
#             s.add(hit)
#             s.flush()
#             s.refresh(hit)
#             logger.info(f"Added {hit.id}")

    def getAvgDurationOfPreviousNHits(self, n) -> int:
        latest_assignments = self.session.query(Assignment).\
          filter(Assignment.created_at!=None).\
          filter(Assignment.submit_page_at!=None).\
          order_by(Assignment.created_at.desc()).limit(n)
        durations = []

        for assignment in latest_assignments:
            durations.append((assignment.submit_page_at - assignment.created_at).total_seconds())
        if not len(durations):
            return int(2.5*60)  # default to 2.5 minutes
        return int(sum(durations) / len(durations))

    def getEstimatedHitDuration(self):
        return self.getAvgDurationOfPreviousNHits(5)

    def getHitTimeout(self):
        return 160 # max(160, self.getAvgDurationOfPreviousNHits(5)*2)

    def getHITs(self, n = 100):
        return self.session.query(HIT).\
                filter(HIT.submit_hit_at != None).\
                order_by(HIT.submit_hit_at.desc()).limit(n)