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 Filename(DeclarativeBase):
  40     __tablename__ = 'filenames'
  41
  42     id = Column(Integer, primary_key=True)
  43     name = Column(String(255), unique=True, index=True)
  44
  45     def __init__(self, name):
  46         self.name = name
  47
  48
  49 class Commit(DeclarativeBase):
  50     __tablename__ = 'commits'
  51
  52     sha1 = Column(String(40), primary_key=True)
  53     cdate = Column(Integer)
  54     adate = Column(Integer)
  55     patch_id = Column(String(40), index=True)
  56     upstream = Column(Boolean)
  57
  58     def __init__(self, sha1, cdate, adate, patch_id, upstream=True):
  59         self.sha1 = sha1
  60         self.cdate = cdate
  61         self.adate = adate
  62         self.patch_id = patch_id
  63         self.upstream = upstream
  64     def __cmp__(self, other):
  65         return cmp(self.cdate, other.cdate)
  66
  67
  68 class Reference(DeclarativeBase):
  69     __tablename__ = 'mailrefs'
  70
  71     id = Column(Integer, primary_key=True)
  72     mail_id = Column(Integer, ForeignKey('mails.id'), index=True)
  73     reference_id = Column(String(255))
  74
  75     def __init__(self, mail_id, reference_id):
  76         self.mail_id = mail_id
  77         self.reference_id = reference_id
  78
  79
  80 class Mail(DeclarativeBase):
  81     __tablename__ = 'mails'
  82
  83     id = Column(Integer, primary_key=True)
  84     message_id = Column(String(255), nullable=False, unique=True)
  85     in_reply_to = Column(String(255))
  86     post_date = Column(Integer)
  87     author = Column(String(255))
  88     subject = Column(String(255))
  89     gmane_id = Column(Integer)
  90     has_patch = Column(Boolean)
  91     stale = Column(Boolean, index=True)
  92     patch_id = Column(String(40), index=True)
  93     data = Column(Binary)
  94
  95     references = relation('Mail', secondary=Reference.__table__,
  96                           primaryjoin='Mail.id==Reference.mail_id',
  97                           secondaryjoin='Mail.message_id==Reference.reference_id',
  98                           foreign_keys=[Reference.mail_id, Reference.reference_id])
  99
 100
 101 class Patch(DeclarativeBase):
 102     __tablename__ = 'patches'
 103
 104     id = Column(Integer, primary_key=True)
 105     commit_sha1 = Column(String(40), ForeignKey('commits.sha1'))
 106     mail_id = Column(Integer, ForeignKey('mails.id'))
 107     extra_notes = Column(Binary)
 108
 109     commit = relation('Commit', backref='patch')
 110     mail = relation('Mail', backref='patch')
 111
 112     def __init__(self, commit, mail_id, extra_notes):
 113         self.commit = commit
 114         self.mail_id = mail_id
 115         self.extra_notes = extra_notes
 116
 117
 118 class Topic(DeclarativeBase):
 119     __tablename__ = 'topics'
 120
 121     id = Column(Integer, primary_key=True)
 122     name = Column(String(255), unique=True, index=True)
 123     timestamp = Column(Integer)
 124     mail_id = Column(Integer, ForeignKey('mails.id'))
 125     cooking_notes = Column(Binary)
 126
 127     mail = relation('Mail', backref='topic')
 128
 129
 130 DeclarativeBase.metadata.create_all(engine)
 131
 132 if __name__ == '__main__':
 133     b = Boundary("aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa")
 134     c = Commit("cccccccccccccccccccccccccccccccccccccccc", 0, 0,
 135                "bbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbb")
 136     b = Blob("aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa")
 137     m = Mail()
 138     p = Patch()