diff options
| -rw-r--r-- | fatcat/models.py | 30 | ||||
| -rw-r--r-- | fatcat/sql.py | 3 | ||||
| -rw-r--r-- | tests/test_backend.py | 2 | 
3 files changed, 9 insertions, 26 deletions
| diff --git a/fatcat/models.py b/fatcat/models.py index 975a5bb7..210d71a6 100644 --- a/fatcat/models.py +++ b/fatcat/models.py @@ -64,7 +64,6 @@ class FileRelease(db.Model):  class WorkRev(db.Model):      __tablename__ = 'work_rev'      id = db.Column(db.Integer, primary_key=True) -    edit_id = db.Column(db.ForeignKey('work_edit.id'))      extra_json = db.Column(db.ForeignKey('extra_json.sha1'), nullable=True)      title = db.Column(db.String) @@ -95,14 +94,15 @@ class WorkEdit(db.Model):      ident_id = db.Column(db.ForeignKey('work_ident.id'), nullable=True)      rev_id = db.Column(db.ForeignKey('work_rev.id'), nullable=True)      redirect_id = db.Column(db.ForeignKey('work_ident.id'), nullable=True) -    edit_group = db.Column(db.ForeignKey('edit_group.id'), nullable=True) +    edit_group_id = db.Column(db.ForeignKey('edit_group.id'), nullable=True)      extra_json = db.Column(db.ForeignKey('extra_json.sha1'), nullable=True) +    rev = db.relationship("WorkRev") +    edit_group = db.relationship("EditGroup")  class ReleaseRev(db.Model):      __tablename__ = 'release_rev'      id = db.Column(db.Integer, primary_key=True, autoincrement=True) -    edit_id = db.Column(db.ForeignKey('release_edit.id'))      extra_json = db.Column(db.ForeignKey('extra_json.sha1'), nullable=True)      work_ident_id = db.ForeignKey('work_ident.id') @@ -135,14 +135,13 @@ class ReleaseEdit(db.Model):      ident_id = db.Column(db.ForeignKey('release_ident.id'), nullable=True)      rev_id = db.Column(db.ForeignKey('release_rev.id'), nullable=True)      redirect_id = db.Column(db.ForeignKey('release_ident.id'), nullable=True) -    edit_group = db.Column(db.ForeignKey('edit_group.id'), nullable=True) +    edit_group_id = db.Column(db.ForeignKey('edit_group.id'), nullable=True)      extra_json = db.Column(db.ForeignKey('extra_json.sha1'), nullable=True)  class CreatorRev(db.Model):      __tablename__ = 'creator_rev'      id = db.Column(db.Integer, primary_key=True, autoincrement=True) -    edit_id = db.Column(db.ForeignKey('creator_edit.id'))      extra_json = db.Column(db.ForeignKey('extra_json.sha1'), nullable=True)      name = db.Column(db.String) @@ -163,14 +162,13 @@ class CreatorEdit(db.Model):      ident_id = db.Column(db.ForeignKey('creator_ident.id'), nullable=True)      rev_id = db.Column(db.ForeignKey('creator_rev.id'), nullable=True)      redirect_id = db.Column(db.ForeignKey('creator_ident.id'), nullable=True) -    edit_group = db.Column(db.ForeignKey('edit_group.id'), nullable=True) +    edit_group_id = db.Column(db.ForeignKey('edit_group.id'), nullable=True)      extra_json = db.Column(db.ForeignKey('extra_json.sha1'), nullable=True)  class ContainerRev(db.Model):      __tablename__ = 'container_rev'      id = db.Column(db.Integer, primary_key=True, autoincrement=True) -    edit_id = db.Column(db.ForeignKey('container_edit.id'))      extra_json = db.Column(db.ForeignKey('extra_json.sha1'), nullable=True)      name = db.Column(db.String) @@ -193,14 +191,13 @@ class ContainerEdit(db.Model):      ident_id = db.Column(db.ForeignKey('container_ident.id'), nullable=True)      rev_id = db.Column(db.ForeignKey('container_rev.id'), nullable=True)      redirect_id = db.Column(db.ForeignKey('container_ident.id'), nullable=True) -    edit_group = db.Column(db.ForeignKey('edit_group.id'), nullable=True) +    edit_group_id = db.Column(db.ForeignKey('edit_group.id'), nullable=True)      extra_json = db.Column(db.ForeignKey('extra_json.sha1'), nullable=True)  class FileRev(db.Model):      __tablename__ = 'file_rev'      id = db.Column(db.Integer, primary_key=True, autoincrement=True) -    edit_id = db.Column(db.ForeignKey('file_edit.id'))      extra_json = db.Column(db.ForeignKey('extra_json.sha1'), nullable=True)      size = db.Column(db.Integer) @@ -222,25 +219,12 @@ class FileEdit(db.Model):      ident_id = db.Column(db.ForeignKey('file_ident.id'), nullable=True)      rev_id = db.Column(db.ForeignKey('file_rev.id'), nullable=True)      redirect_id = db.Column(db.ForeignKey('file_ident.id'), nullable=True) -    edit_group = db.Column(db.ForeignKey('edit_group.id'), nullable=True) +    edit_group_id = db.Column(db.ForeignKey('edit_group.id'), nullable=True)      extra_json = db.Column(db.ForeignKey('extra_json.sha1'), nullable=True)  ### Editing ################################################################# -#class Edit(db.Model): -#    __tablename__ = 'edit' -#    id = db.Column(db.Integer, primary_key=True, autoincrement=True) -#    edit_group = db.Column(db.ForeignKey('edit_group.id'), nullable=True) -#    editor = db.Column(db.ForeignKey('editor.id'), nullable=False) -#    comment = db.Column(db.String, nullable=True) -#    extra_json = db.Column(db.ForeignKey('extra_json.sha1'), nullable=True) -#    # WARNING: polymorphic. Represents the ident that should end up pointing to -#    # this revision. -#    entity_ident = db.Column(db.Integer, nullable=True) -#    entity_rev = db.Column(db.Integer, nullable=True) -#    entity_redirect = db.Column(db.Integer, nullable=True) -  class EditGroup(db.Model):      __tablename__ = 'edit_group'      id = db.Column(db.Integer, primary_key=True, autoincrement=True) diff --git a/fatcat/sql.py b/fatcat/sql.py index 782f7983..6ca56b46 100644 --- a/fatcat/sql.py +++ b/fatcat/sql.py @@ -228,7 +228,6 @@ def hydrate_work(wid):      hydro.update({          "work_type": work.revision.work_type,          "title": work.revision.title, -        "edit_id": work.revision.edit_id,          "primary": primary,          "creators": creators,          "releases": releases, @@ -244,7 +243,7 @@ def hydrate_release(rid):          "_type": "release",          "id": rid,          "revision": release.rev_id, -        "edit_id": release.revision.edit_id, +        #"edit_id": release.revision.edit_id,          "is_live": release.live,          "work_id": release.revision.work_ident_id, diff --git a/tests/test_backend.py b/tests/test_backend.py index bc610d25..93e3f11a 100644 --- a/tests/test_backend.py +++ b/tests/test_backend.py @@ -12,7 +12,7 @@ import tempfile  ## Helpers ##################################################################  def check_entity_fields(e): -    for key in ('id', 'rev', 'redirect_id', 'edit_id'): +    for key in ('id', 'rev', 'redirect_id'):          assert key in e      for key in ('id',):          assert e[key] is not None | 
