db.py

   1 from sqlalchemy.ext.declarative import declarative_base
   2 from sqlalchemy import Table, Column, Integer, String, Binary, Boolean, \
   3     ForeignKey, create_engine
   4 from sqlalchemy.orm import sessionmaker
   5 from sqlalchemy.orm import relation, backref
   6
   7 DeclarativeBase = declarative_base()
   8
   9 engine = create_engine('postgres://trackgit:trackgit@localhost/trackgit', echo=False)
  10 Session = sessionmaker(bind=engine)
  11
  12
  13 class Boundary(DeclarativeBase):
  14     __tablename__ = 'boundaries'
  15     sha1 = Column(String(40), primary_key=True)
  16     def __init__(self, sha1):
  17         self.sha1 = sha1
  18
  19
  20 class Blob(DeclarativeBase):
  21     __tablename__ = 'blobs'
  22
  23     sha1 = Column(String(40), primary_key=True)
  24     newest_commit_sha1 = Column(String(40), ForeignKey('commits.sha1'))
  25     oldest_commit_sha1 = Column(String(40), ForeignKey('commits.sha1'))
  26
  27     newest_commit = relation('Commit', primaryjoin='Blob.newest_commit_sha1==Commit.sha1')
  28     oldest_commit = relation('Commit', primaryjoin='Blob.oldest_commit_sha1==Commit.sha1')
  29
  30     def __init__(self, sha1):
  31         self.sha1 = sha1
  32     def update_contained_in(self, commit):
  33         #if not self.oldest_commit or self.oldest_commit.cdate > commit.cdate:
  34         #    self.oldest_commit = commit
  35         if not self.newest_commit or self.newest_commit.cdate < commit.cdate:
  36             self.newest_commit = commit
  37
  38
  39 class Commit(DeclarativeBase):
  40     __tablename__ = 'commits'
  41
  42     sha1 = Column(String(40), primary_key=True)
  43     cdate = Column(Integer)
  44     adate = Column(Integer)
  45     patch_id = Column(String(40))
  46     upstream = Column(Boolean)
  47
  48     def __init__(self, sha1, cdate, adate, patch_id, upstream=True):
  49         self.sha1 = sha1
  50         self.cdate = cdate
  51         self.adate = adate
  52         self.patch_id = patch_id
  53         self.upstream = upstream
  54     def __cmp__(self, other):
  55         return cmp(self.cdate, other.cdate)
  56
  57
  58 class Reference(DeclarativeBase):
  59     __tablename__ = 'references'
  60
  61     id = Column(Integer, primary_key=True)
  62     mail_id = Column(Integer, ForeignKey('mails.id'))
  63     reference_id = Column(String(255))
  64
  65     def __init__(self, mail_id, reference_id):
  66         self.mail_id = mail_id
  67         self.reference_id = reference_id
  68
  69
  70 class Mail(DeclarativeBase):
  71     __tablename__ = 'mails'
  72
  73     id = Column(Integer, primary_key=True)
  74     message_id = Column(String(255), nullable=False, unique=True)
  75     in_reply_to = Column(String(255))
  76     post_date = Column(Integer)
  77     author = Column(String(255))
  78     subject = Column(String(255))
  79     gmane_id = Column(Integer)
  80     has_patch = Column(Boolean)
  81     stale = Column(Boolean)
  82     patch_id = Column(String(40))
  83     data = Column(Binary)
  84
  85     references = relation('Mail', secondary=Reference.__table__,
  86                           primaryjoin='Mail.id==Reference.mail_id',
  87                           secondaryjoin='Mail.message_id==Reference.reference_id',
  88                           foreign_keys=[Reference.mail_id, Reference.reference_id])
  89
  90
  91 class Patch(DeclarativeBase):
  92     __tablename__ = 'patches'
  93
  94     id = Column(Integer, primary_key=True)
  95     commit_sha1 = Column(String(40), ForeignKey('commits.sha1'))
  96     mail_id = Column(Integer, ForeignKey('mails.id'))
  97     extra_notes = Column(Binary)
  98
  99     commit = relation('Commit', backref='patch')
 100     mail = relation('Mail', backref='patch')
 101
 102     def __init__(self, commit, mail_id, extra_notes):
 103         self.commit = commit
 104         self.mail_id = mail_id
 105         self.extra_notes = extra_notes
 106
 107
 108 class Topic(DeclarativeBase):
 109     __tablename__ = 'topics'
 110
 111     id = Column(Integer, primary_key=True)
 112     name = Column(String(255))
 113     mail_id = Column(Integer, ForeignKey('mails.id'))
 114     cooking_notes = Column(Binary)
 115
 116     mail = relation('Mail', backref='topic')
 117
 118
 119 DeclarativeBase.metadata.create_all(engine)
 120
 121 if __name__ == '__main__':
 122     b = Boundary("aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa")
 123     c = Commit("cccccccccccccccccccccccccccccccccccccccc", 0, 0,
 124                "bbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbb")
 125     b = Blob("aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa")
 126     m = Mail()
 127     p = Patch()