WIP

1 files changed, 126 insertions, 4 deletions

diff --git a/backend/backend.py b/backend/backend.py
index 8f9322ab..a39ae790 100755
--- a/backend/backend.py
+++ b/backend/backend.py
@@ -5,7 +5,16 @@ from flask import Flask, render_template, send_from_directory, request, \
 from sqlalchemy import create_engine, MetaData, Table
 
 app = Flask(__name__)
-app.config.from_pyfile('config.py')
+app.config.from_object(__name__)
+
+# Load default config and override config from an environment variable
+app.config.update(dict(
+    DATABASE_URI='sqlite://:memory:',
+    SECRET_KEY='development-key',
+    USERNAME='admin',
+    PASSWORD='admin'
+))
+app.config.from_envvar('FATCAT_BACKEND_CONFIG', silent=True)
 
 metadata = MetaData()
 
@@ -54,7 +63,7 @@ release_revision = Table('release_revision', metadata,
     Column('edit_id', ForeignKey('edit.id')),
     Column('extra_json', ForeignKey('extra_json.sha1'), nullable=True),
 
-    Column('work', ForeignKey('work_id.id')),
+    #Column('work', ForeignKey('work_id.id')),
     Column('container', ForeignKey('container_id.id')),
     Column('title', String),
     Column('license', String),          # TODO: oa status foreign key
@@ -114,6 +123,30 @@ container_revision = Table('container_revision', metadata,
     Column('issn', String),             # TODO: identifier table
     )
 
+file_id = Table('file_id', metadata,
+    Column('id', Integer, primary_key=True, autoincrement=False),
+    Column('revision', ForeignKey('container_revision.id')),
+    )
+
+file_revision = Table('file_revision', metadata,
+    Column('id', Integer, primary_key=True, autoincrement=True),
+    Column('previous', ForeignKey('file_revision.id'), nullable=True),
+    Column('state', Enum(IdState)),
+    Column('redirect_id', ForeignKey('file_id.id'), nullable=True),
+    Column('edit_id', ForeignKey('edit.id')),
+    Column('extra_json', ForeignKey('extra_json.sha1'), nullable=True),
+
+    Column('size', Integer),
+    Column('sha1', Integer),            # TODO: hash table... only or in addition?
+    Column('url', Integer),             # TODO: URL table
+    )
+
+release_file= Table('release_file', metadata,
+    Column('id', Integer, primary_key=True, autoincrement=True),
+    Column('release_rev', ForeignKey('release_revision.id'), nullable=False),
+    Column('file_id', ForeignKey('file_id.id'), nullable=False),
+    )
+
 edit = Table('edit', metadata,
     Column('id', Integer, primary_key=True, autoincrement=True),
     Column('edit_group', ForeignKey('edit_group.id')),
@@ -143,6 +176,82 @@ extra_json = Table('extra_json', metadata,
     Column('json', String),
     )
 
+## Helpers ##################################################################
+
+def is_fcid(s):
+    return len(s) == 26 and s.isalnum()
+
+# XXX: why isn't this running?
+def test_is_fcid():
+
+    for s in ("rzga5b9cd7efgh04iljk", "RZGA5B9CD7Efgh04iljk"):
+        assert is_fcid() is True
+
+    for s in ("rzga59cd7efgh04iljk", "rzga.b9cd7efgh04iljk", "", 
+            "rzga5b9cd7efgh04iljkz"):
+        assert is_fcid() is False
+
+def release_list(id_list):
+    # XXX: MOCK
+    l = []
+    for i in id_list:
+        l.append({
+            "id": i,
+            "rev": "8fkj28fjhqkjdhkjkj9s",
+            "previous": "0021jdfjhqkjdhkjkj9s",
+            "state": "normal",
+            "redirect_id": None,
+            "edit_id": "932582iuhckjvssk",
+            "extra_json": None,
+
+            "container_id": "0021jdfjhqkjdhkjkj9s",
+            "title": "Mocks are great",
+            "license": "CC-0",
+            "release_type": "publication",
+            "date": "2017-11-22",
+            "doi": "10.1000/953kj.sdfkj",
+        })
+    return l
+
+def release_hydrate(release_id):
+    e = release_list([release_id])[0]
+    e['container'] = container_hydrate(d['container_id'])
+    e.pop('container_id')
+    e['creators'] = [creator_hydrate(c['id']) for c in e['creator_ids']]
+    return e
+
+def work_list(id_list):
+    """This is the fast/light version: populates entity-specific lists (eg,
+    identifiers), and any primaries, but doesn't transclude all other
+    entities"""
+    if len(id_list) == 0:
+        return []
+
+    l = []
+    for i in id_list:
+        l.append({
+            "id": "rzga5b9cd7efgh04iljk",
+            "rev": "8fkj28fjhqkjdhkjkj9s",
+            "previous": "0021jdfjhqkjdhkjkj9s",
+            "state": "normal",
+            "redirect_id": None,
+            "edit_id": "932582iuhckjvssk",
+            "extra_json": None,
+
+            "title": "Mocks are great",
+            "contributors": [],
+            "work_type": "journal-article",
+            "date": None,
+
+            "primary_release": release_list(["8fkj28fjhqkjdhkjkj9s"])[0],
+        })
+    return l
+
+def work_hydrate(work_id):
+    """This is the heavy/slowversion: everything from get_works(), but also
+    recursively transcludes single-linked entities"""
+    # XXX:
+    return work_list([work_id])[0]
 
 ## API Methods ##############################################################
 
@@ -150,6 +259,15 @@ extra_json = Table('extra_json', metadata,
 def health():
     return jsonify({'ok': True})
 
+
+@app.route('/v0/work/<work_id>', methods=['GET'])
+def work_get(work_id):
+    if not is_fcid(work_id):
+        print("not fcid: {}".format(work_id))
+        return abort(404)
+    work = work_hydrate(work_id)
+    return jsonify(work)
+
 ## Entry Point ##############################################################
 
 def main():
@@ -165,13 +283,17 @@ def main():
         default=8040,
         help="listen on this port")
     parser.add_argument('--database-uri',
-        default="sqlite:///test.sqlite",
+        default=app.config['DATABASE_URI'],
         help="sqlalchemy database string")
     args = parser.parse_args()
 
     app.config['DATABASE_URI'] = args.database_uri
-    engine = create_engine(app.config['DATABASE_URI'], convert_unicode=True)
+    app.conn = create_engine(app.config['DATABASE_URI'], convert_unicode=True)
     metadata.create_all(bind=engine)
+
+    # XXX:
+    db_test_data()
+
     app.run(debug=args.debug, host=args.host, port=args.port)