git-importer/lib/db_revision.py

class DBRevision:
    def __init__(self, row):
        # need to stay in sync with the schema creation in db.py
        (
            self.dbid,
            self.project,
            self.package,
            self.rev,
            self.unexpanded_srcmd5,
            self.commit_time,
            self.userid,
            self.comment,
            self.requestid,
        ) = row

    def links_to(self, db, project, package):
        db.cursor().execute(
            "INSERT INTO links (revision_id, project, package) VALUES (%s,%s,%s)",
            (self.dbid, project, package),
        )
        print("links_to", project, package)

    @classmethod
    def import_obs_rev(cls, db, revision):
        cur = db.cursor()
        cur.execute(
            """INSERT INTO revisions (project, package, rev, unexpanded_srcmd5, commit_time, userid, comment, requestid)  
                    VALUES(%s, %s, %s, %s, %s, %s, %s, %s)""",
            (
                revision.project,
                revision.package,
                revision.rev,
                revision.unexpanded_srcmd5,
                revision.time,
                revision.userid,
                revision.comment,
                revision.requestid,
            ),
        )
        cur.close()
        return cls.fetch_revision(db, revision.project, revision.package, revision.rev)

    @classmethod
    def fetch_revision(cls, db, project, package, rev):
        cur = db.cursor()
        cur.execute(
            "SELECT * FROM revisions where project=%s and package=%s and rev=%s",
            (project, package, str(rev)),
        )
        row = cur.fetchone()
        cur.close()
        return DBRevision(row)

    @classmethod
    def latest_revision(cls, db, project, package):
        cur = db.cursor()
        cur.execute(
            "SELECT MAX(rev) FROM revisions where project=%s and package=%s",
            (project, package),
        )
        max = cur.fetchone()[0]
        cur.close()
        if max:
            return DBRevision.fetch_revision(db, project, package, int(max))
        return None