/
database.py
76 lines (58 loc) · 1.86 KB
/
database.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
#!/usr/bin/env python
# -*- coding: utf-8 -*-
"""
knuth.database
~~~~~~~~~~~~~~
Database interaction for the knuth search engine.
(C) 10.12.25, 2013, 2014, Lukas Prokop, Andi
"""
from flask.ext.sqlalchemy import SQLAlchemy
from datetime import datetime
import time
db = SQLAlchemy()
class Document(db.Model):
"""A document like papers, theses, ..."""
__tablename__ = "documents"
# table columns
id = db.Column(db.Integer, primary_key=True, autoincrement=True)
type = db.Column(db.String(20))
title = db.Column(db.String(256))
author = db.Column(db.String(40))
doi = db.Column(db.String(256))
timestamp = db.Column(db.Integer)
parent = db.Column(db.Integer)
def __init__(self, type='doc', title='', author='',
doi='', timestamp=None, parent=None):
if not timestamp:
timestamp = int(time.time())
self.type = type
self.title = title
self.author = author
self.doi = doi
self.timestamp = timestamp
self.parent = parent
def __repr__(self):
return 'Document({})'.format(self.title)
def get_date(self):
if self.timestamp == 0:
return None
return datetime.utcfromtimestamp(self.timestamp)
def get_format_date_string(self):
dt = self.get_date()
if dt:
return dt.strftime('%Y-%m-%d')
else:
return ''
class Metadata(db.Model):
"""Table for metadata of a document"""
__tablename__ = "metadata"
# table columns
document = db.Column(db.Integer, primary_key=True)
key = db.Column(db.String(40), primary_key=True)
value = db.Column(db.Text, primary_key=True)
def __init__(self, document, key, value=''):
self.document = document
self.key = key
self.value = value
def __repr__(self):
return self.key