IDSIA / sacred

Base = declarative_base()

class Source(Base):

    __tablename__ = 'source'

    @classmethod

    def get_or_create(cls, filename, md5sum, session):

        instance = session.query(cls).filter_by(filename=filename,

                                                md5sum=md5sum).first()

        if instance:

            return instance

        md5sum_ = get_digest(filename)

        assert md5sum_ == md5sum, 'Weird: found md5 mismatch for {}: {} != {}'\

            .format(filename, md5sum, md5sum_)

        with open(filename, 'r') as f:

            return cls(filename=filename, md5sum=md5sum, content=f.read())

    source_id = sa.Column(sa.Integer, primary_key=True)

    filename = sa.Column(sa.String(256))

    md5sum = sa.Column(sa.String(32))

    content = sa.Column(sa.Text)

    def to_json(self):

        return {'filename': self.filename,

                'md5sum': self.md5sum}

class Dependency(Base):

                'filename': self.filename}

class Resource(Base):

    __tablename__ = 'resource'

    @classmethod

    def get_or_create(cls, filename, session):

        md5sum = get_digest(filename)

        instance = session.query(cls).filter_by(filename=filename,

                                                md5sum=md5sum).first()

        if instance:

            return instance

        with open(filename, 'rb') as f:

            return cls(filename=filename, md5sum=md5sum, content=f.read())

    resource_id = sa.Column(sa.Integer, primary_key=True)

    filename = sa.Column(sa.String(256))

    md5sum = sa.Column(sa.String(32))

    content = sa.Column(sa.LargeBinary)

    def to_json(self):

        return {'filename': self.filename,

                'md5sum': self.md5sum}

class Host(Base):