db.py 8.67 KB
Newer Older
1
2
# -*- coding: utf-8 -*-

3
import defs
4
from gluon import current
5

6
7
# AUTODELETE should be set to False before any maintenance operation on the DB
AUTODELETE = True
8

9
10
11
# Length of the upload field
LENGTH_UPLOAD = 400

12
13
14
15
16
17
18
#########################################################################
## This scaffolding model makes your app work on Google App Engine too
## File is released under public domain and you can use without limitations
#########################################################################

## if SSL/HTTPS is properly configured and you want all HTTP requests to
## be redirected to HTTPS, uncomment the line below:
19
request.requires_https()
20
21
22

if not request.env.web2py_runtime_gae:
    ## if NOT running on Google App Engine use SQLite or other DB
23
    db = DAL(defs.DB_ADDRESS,pool_size=1,check_reserved=['all'])
24
    ## db = DAL('mysql://root:password@localhost/vidjil5',pool_size=1,check_reserved=['all'])
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
else:
    ## connect to Google BigTable (optional 'google:datastore://namespace')
    db = DAL('google:datastore')
    ## store sessions and tickets there
    session.connect(request, response, db=db)
    ## or store session in Memcache, Redis, etc.
    ## from gluon.contrib.memdb import MEMDB
    ## from google.appengine.api.memcache import Client
    ## session.connect(request, response, db = MEMDB(Client()))

## by default give a view/generic.extension to all actions from localhost
## none otherwise. a pattern can be 'controller/function.extension'
response.generic_patterns = ['*'] if request.is_local else []
## (optional) optimize handling of static files
# response.optimize_css = 'concat,minify,inline'
# response.optimize_js = 'concat,minify,inline'

#########################################################################
## Here is sample code if you need for
## - email capabilities
## - authentication (registration, login, logout, ... )
## - authorization (role based authorization)
## - services (xml, csv, json, xmlrpc, jsonrpc, amf, rss)
## - old style crud actions
## (more options discussed in gluon/tools.py)
#########################################################################

from gluon.tools import Auth, Crud, Service, PluginManager, prettydate
auth = Auth(db)
crud, service, plugins = Crud(db), Service(), PluginManager()

## create all tables needed by auth if not custom tables
auth.define_tables(username=False, signature=False)

## configure email
mail = auth.settings.mailer
mail.settings.server = 'logging' or 'smtp.gmail.com:587'
mail.settings.sender = 'you@gmail.com'
mail.settings.login = 'username:password'

## configure auth policy
auth.settings.registration_requires_verification = False
auth.settings.registration_requires_approval = False
auth.settings.reset_password_requires_verification = True

70
71
72
# auth.settings.login_email_validate = False
auth.settings.expiration = 3600 * 24 * 7  # one week
auth.settings.remember_me_form = False
73
74
auth.settings.logged_url = URL('user', 'info')
auth.settings.login_next = URL('user', 'info')
75

76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
## if you need to use OpenID, Facebook, MySpace, Twitter, Linkedin, etc.
## register with janrain.com, write your domain:api_key in private/janrain.key
from gluon.contrib.login_methods.rpx_account import use_janrain
use_janrain(auth, filename='private/janrain.key')

#########################################################################
## Define your tables below for example
##
## >>> db.define_table('mytable',Field('myfield','string'))
##
## Fields can be 'string','text','password','integer','double','boolean'
##       'date','time','datetime','blob','upload', 'reference TABLENAME'
## There is an implicit 'id integer autoincrement' field
## Consult manual for more options, validators, etc.
##
## More API examples for controllers:
##
## >>> db.mytable.insert(myfield='value')
## >>> rows=db(db.mytable.myfield=='value').select(db.mytable.ALL)
## >>> for row in rows: print row.id, row.myfield
#########################################################################

db.define_table('patient',
                Field('first_name','string'),
                Field('last_name','string'),
                Field('birth','date'),
102
103
                Field('info','text'),
                Field('id_label','string'),
104
                Field('creator','reference auth_user'))
105
106
107
108
109
110
111
112
113
114
115
116
117
118

'''
db.patient.first_name.requires = IS_NOT_EMPTY( error_message='input needed' )
db.patient.last_name.requires = IS_NOT_EMPTY( error_message='input needed' )
db.patient.birth.requires = IS_DATE(format=T('%Y-%m-%d'),
                   error_message='must be YYYY-MM-DD!')
'''



db.define_table('sequence_file',
                Field('patient_id', 'reference patient'),
                Field('sampling_date','date'),
                Field('info','text'),
Marc Duez's avatar
Marc Duez committed
119
                Field('filename','text'),
120
121
122
                Field('pcr','text'),
                Field('sequencer','text'),
                Field('producer','text'),
123
124
                Field('size_file','integer', default=0),
                Field('provider','reference auth_user'),
125
                Field('data_file', 'upload', 
126
                      uploadfolder=defs.DIR_SEQUENCES,
127
                      length=LENGTH_UPLOAD, autodelete=AUTODELETE))
128
129
130
131
132
133
134




db.define_table('standard_file',
                Field('name', 'string'),
                Field('info','text'),
135
                Field('data_file', 'upload',
136
                      uploadfolder=defs.DIR_SEQUENCES,
137
                      autodelete=AUTODELETE, length=LENGTH_UPLOAD))
138
139
140
141
142



db.define_table('config',
                Field('name', 'string'),
143
                Field('program', 'string'),
144
                Field('command', 'string'),
145
                Field('fuse_command', 'string'),
146
                Field('info','text'))
147
148


149
db.define_table('results_file',
150
151
                Field('sequence_file_id', 'reference sequence_file'),
                Field('config_id', 'reference config'),
152
                Field('run_date','datetime'),
153
                Field('scheduler_task_id', 'integer'),
154
                Field('data_file', 'upload', 
155
                      uploadfolder=defs.DIR_RESULTS,
156
                      length=LENGTH_UPLOAD, autodelete=AUTODELETE))
157

158
159
160
db.define_table('fused_file',
                Field('patient_id', 'reference patient'),
                Field('config_id', 'reference config'),
161
                Field('fuse_date','datetime', default="1970-01-01 00:00:00"),
162
                Field('status', 'string'),
163
                Field('sequence_file_list', 'string'),
164
                Field('fused_file', 'upload', 
165
                      uploadfolder=defs.DIR_RESULTS,
166
                      length=LENGTH_UPLOAD, autodelete=AUTODELETE))
167
168
169
170

db.define_table('analysis_file',
                Field('patient_id', 'reference patient'),
                Field('config_id', 'reference config'),
171
                Field('analyze_date','datetime'),
172
                Field('status', 'string'),
173
                Field('analysis_file', 'upload', 
174
                      uploadfolder=defs.DIR_RESULTS,
175
                      length=LENGTH_UPLOAD, autodelete=AUTODELETE))
176

177
178
179

## after defining tables, uncomment below to enable auditing
auth.enable_record_versioning(db)
180

181
182
183
184
185
## Reverse IP

ips = {}

try:
186
    for l in open(defs.REVERSE_IP):
187
188
189
190
191
192
        ip, kw = l.split()
        ips[ip] = kw
except:
    pass


193
194
195
196
## Logging

import logging

197
198
199
logging.ADMIN = logging.INFO + 1
logging.addLevelName(logging.ADMIN, 'ADMIN')

200
201
202
203
class MsgUserAdapter(logging.LoggerAdapter):

    def process(self, msg, kwargs):
        if type(msg) is dict:
204
205
206
207
            if 'message' in msg:
                msg = msg['message']
            else:
                msg = '?'
208
        ip = request.client
209
210
211
212
        if ip:
            for ip_prefix in ips:
                if ip.startswith(ip_prefix):
                    ip = "%s/%s" % (ip, ips[ip_prefix])
213
        new_msg =  '%30s %12s %s' % (ip, ('<%s>' % auth.user.first_name.replace(' ','-') if auth.user else ''), msg)
214
215
        return new_msg, kwargs
    
216
217
    def admin(self, msg):
        self.log(logging.ADMIN, msg)
218
219
220
221
222
223
224
225
226
227
228
229
#

def _init_log():
    """
    adapted from http://article.gmane.org/gmane.comp.python.web2py/11091
    """

    import logging

    logger = logging.getLogger('vidjil') # (request.application)
    if not logger.handlers:
        logger.setLevel(logging.DEBUG)
230
231
        formatter = logging.Formatter('[%(process)d] %(asctime)s %(levelname)8s - %(filename)s:%(lineno)d\t%(message)s')

232
        handler = logging.FileHandler(defs.LOG_DEBUG)
233
        handler.setLevel(logging.DEBUG)
234
235
236
        handler.setFormatter(formatter)
        logger.addHandler(handler) 

237
        handler = logging.FileHandler(defs.LOG_INFO)
238
239
        handler.setLevel(logging.INFO)
        handler.setFormatter(formatter)
240
        logger.addHandler(handler) 
241

242
        logger.debug("Creating logger")
243
244
245
    return MsgUserAdapter(logger, {})

log = _init_log()
246

Marc Duez's avatar
Marc Duez committed
247
current.log = log
248
249
current.db = db
current.auth = auth