Mental-Cache/index.py at master · ryanrdetzel/Mental-Cache · GitHub

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
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
102
103
104
105
106
107
108
109
110
111
112
113
#!/usr/bin/env python
import web
import page
import upload
import utils
import logging
import pickle
from Cheetah.Template import Template
import os
import re
import pytc
import simplejson as json

urls = (
    '/page', page.app_page,
    '/upload', upload.app_upload,
    '/login', "login",
    '/logout', "logout",
    '/(\w+)\/(?:[\w|-]+)\.html', "index",
    '/', "dashboard",
    "/(.*)", "default"
)

class login:
    def GET(self):
        path = os.path.join(os.path.dirname(__file__), 'templates/login.html')
        template_values = {}
        tmpl = Template( file = path, searchList = (template_values,) )
        return tmpl

    def POST(self):
        ## Check if they are valid
        data = web.input(email="")

        db = pytc.HDB()
        db.open(utils.DBNAME, pytc.HDBOWRITER | pytc.HDBOCREAT)

        if db.has_key(data.email):
            profile = pickle.loads(db.get(data.email))
            session.profile = profile
            session.userid = profile['id']

        raise web.seeother('/')

class logout:
    def GET(self):
        session.kill()
        raise web.seeother('/login')

class dashboard:
    def GET(self):
        if session.userid is not None:
            ## This should come from an index but this is something to hold me over.

            db = pytc.HDB()
            db.open(utils.DBNAME, pytc.HDBOWRITER | pytc.HDBOCREAT)

            pages = []
            if db.has_key(session.userid + ':index'):
                content = db.get(session.userid + ':index')
                pages = json.loads(content)

            path = os.path.join(os.path.dirname(__file__), 'templates/dashboard.html')
            template_values = { 'pages' : pages }
            tmpl = Template( file = path, searchList = (template_values,) )
            return tmpl
        raise web.seeother('/login')


class default:
    def GET(self,what):
        return 'page not found'


class index:
    def GET(self,page_name):
        if session.userid is not None:
            ## Validate that this user can view this page
            if utils.check_permissions(page_name,session.userid):
                path = os.path.join(os.path.dirname(__file__), 'templates/index.html')
                ## Check to see if there a public permission
                template_values = { 'page_name':page_name, 'profile' : session['profile'], 'can_write' : web.config._canwrite, 'is_public': web.config._ispublic }
            else:
                path = os.path.join(os.path.dirname(__file__), 'templates/denied.html')
                template_values = {}

            tmpl = Template( file = path, searchList = (template_values,) )
            return tmpl
        else:
   #         path = os.path.join(os.path.dirname(__file__), 'templates/login.html')
   #         template_values = {}
            raise web.seeother('/login')


app = web.application(urls, globals())

web.config.debug = False

web.config.session_parameters['cookie_name'] = 'mentalcache'
web.config.session_parameters['cookie_domain'] = None
web.config.session_parameters['timeout'] = 86400, #24 * 60 * 60, # 24 hours   in seconds
#web.config.session_parameters['timeout'] = 30
web.config.session_parameters['ignore_change_ip'] = True
web.config.session_parameters['secret_key'] = 'fLjU209834kjhsdf8213'
web.config.session_parameters['expired_message'] = 'Session expired'

session = web.session.Session(app, web.session.DiskStore('/tmp/sessions'), initializer={'userid': None})

web.config._session = session

if __name__ == '__main__':
    logging.getLogger().setLevel(logging.DEBUG)
    app.run()