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

3
import defs
4
from gluon import current
5

6 7 8 9 10 11 12
#########################################################################
## 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:
13
request.requires_https()
14 15 16 17

if not request.env.web2py_runtime_gae:
    ## if NOT running on Google App Engine use SQLite or other DB
    db = DAL('sqlite://storage.sqlite',pool_size=1,check_reserved=['all'])
18
    ## db = DAL('mysql://root:password@localhost/vidjil5',pool_size=1,check_reserved=['all'])
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
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

64 65 66
# auth.settings.login_email_validate = False
auth.settings.expiration = 3600 * 24 * 7  # one week
auth.settings.remember_me_form = False
67 68
auth.settings.logged_url = URL('user', 'info')
auth.settings.login_next = URL('user', 'info')
69

70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95
## 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'),
96 97
                Field('info','text'),
                Field('id_label','string'),
98
                Field('creator','reference auth_user'))
99 100 101 102 103 104 105 106 107 108 109 110 111 112

'''
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
113
                Field('filename','text'),
114 115 116
                Field('pcr','text'),
                Field('sequencer','text'),
                Field('producer','text'),
117 118
                Field('size_file','integer', default=0),
                Field('provider','reference auth_user'),
119
                Field('data_file', 'upload', 
120
                      uploadfolder=defs.DIR_SEQUENCES,
121
                      length=4294967295, autodelete=True))
122 123 124 125 126 127 128




db.define_table('standard_file',
                Field('name', 'string'),
                Field('info','text'),
129
                Field('data_file', 'upload',
130
                      uploadfolder=defs.DIR_SEQUENCES,
131
                      autodelete=True, length=4294967295))
132 133 134 135 136



db.define_table('config',
                Field('name', 'string'),
137
                Field('program', 'string'),
138
                Field('command', 'string'),
139
                Field('fuse_command', 'string'),
140
                Field('info','text'))
141 142


143
db.define_table('results_file',
144 145
                Field('sequence_file_id', 'reference sequence_file'),
                Field('config_id', 'reference config'),
146
                Field('run_date','datetime'),
147
                Field('scheduler_task_id', 'integer'),
148
                Field('data_file', 'upload', 
149
                      uploadfolder=defs.DIR_RESULTS,
150
                      length=4294967295, autodelete=True))
151

152 153 154
db.define_table('fused_file',
                Field('patient_id', 'reference patient'),
                Field('config_id', 'reference config'),
155
                Field('fuse_date','datetime', default="1970-01-01 00:00:00"),
156
                Field('status', 'string'),
157
                Field('sequence_file_list', 'string'),
158
                Field('fused_file', 'upload', 
159
                      uploadfolder=defs.DIR_RESULTS,
160
                      length=4294967295, autodelete=True))
161 162 163 164

db.define_table('analysis_file',
                Field('patient_id', 'reference patient'),
                Field('config_id', 'reference config'),
165
                Field('analyze_date','datetime'),
166
                Field('status', 'string'),
167
                Field('analysis_file', 'upload', 
168
                      uploadfolder=defs.DIR_RESULTS,
169 170
                      length=4294967295, autodelete=True))

171 172 173 174


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

176 177 178 179 180
## Reverse IP

ips = {}

try:
181
    for l in open(defs.REVERSE_IP):
182 183 184 185 186 187
        ip, kw = l.split()
        ips[ip] = kw
except:
    pass


188 189 190 191 192 193 194 195
## Logging

import logging

class MsgUserAdapter(logging.LoggerAdapter):

    def process(self, msg, kwargs):
        if type(msg) is dict:
196 197 198 199
            if 'message' in msg:
                msg = msg['message']
            else:
                msg = '?'
200
        ip = request.client
201 202 203 204 205
        if ip:
            for ip_prefix in ips:
                if ip.startswith(ip_prefix):
                    ip = "%s/%s" % (ip, ips[ip_prefix])
        new_msg =  '%30s %12s %s' % (ip, ('<%s>' % auth.user.first_name if auth.user else ''), msg)
206 207 208 209 210 211 212 213 214 215 216 217 218 219
        return new_msg, kwargs
    
#

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)
220 221
        formatter = logging.Formatter('[%(process)d] %(asctime)s %(levelname)8s - %(filename)s:%(lineno)d\t%(message)s')

222
        handler = logging.FileHandler(defs.LOG_DEBUG)
223
        handler.setLevel(logging.DEBUG)
224 225 226
        handler.setFormatter(formatter)
        logger.addHandler(handler) 

227
        handler = logging.FileHandler(defs.LOG_INFO)
228 229
        handler.setLevel(logging.INFO)
        handler.setFormatter(formatter)
230
        logger.addHandler(handler) 
231

232
        logger.debug("Creating logger")
233 234 235
    return MsgUserAdapter(logger, {})

log = _init_log()
236

Marc Duez's avatar
Marc Duez committed
237
current.log = log
238 239
current.db = db
current.auth = auth