52 lines
1.3 KiB
Python
52 lines
1.3 KiB
Python
"""
|
|
This file contains the database schema in SQLAlchemy format.
|
|
"""
|
|
from sqlalchemy import event, Column, Integer, String
|
|
from sqlalchemy.engine import Engine
|
|
from sqlalchemy.ext.declarative import declarative_base
|
|
|
|
Base = declarative_base()
|
|
|
|
|
|
@event.listens_for(Engine, "connect")
|
|
def set_sqlite_pragma(dbapi_connection, connection_record):
|
|
"""
|
|
Auto enable foreign keys for SQLite.
|
|
"""
|
|
cursor = dbapi_connection.cursor()
|
|
cursor.execute("PRAGMA foreign_keys=ON")
|
|
cursor.close()
|
|
|
|
|
|
class Paper(Base):
|
|
__tablename__ = 'papers'
|
|
id = Column(Integer, primary_key=True)
|
|
doi = Column(String(), nullable=True, unique=True)
|
|
arxiv_id = Column(String(25), nullable=True, unique=True)
|
|
|
|
def __repr__(self):
|
|
return "<Paper(id='%d', doi='%s', arxiv_id='%s')>" % (
|
|
self.id,
|
|
self.doi,
|
|
self.arxiv_id,
|
|
)
|
|
|
|
def json_api_repr(self):
|
|
"""
|
|
Dict to dump for the JSON API.
|
|
"""
|
|
return {
|
|
"types": self.__tablename__,
|
|
"id": self.id,
|
|
"attributes": {
|
|
"doi": self.doi,
|
|
"arxiv_id": self.arxiv_id,
|
|
},
|
|
"links": {
|
|
"self": "/papers/%d" % (self.id,)
|
|
},
|
|
"relationships": {
|
|
# TODO
|
|
}
|
|
}
|