コード例 #1
0
ファイル: controller.py プロジェクト: Tyler-D/Zfile
def is_admin():
    if not is_user():
        return False

    token = web.ctx.session.token
    userid = auth.decode_token(token)['userid']
    db = MySQLDB_(settings['MYSQL_HOST'], settings['MYSQL_USER'],
            settings['MYSQL_PASSWD'], settings['MYSQL_DB'])
    user_li = db.query_all("select * from user where `id`='%s'" % userid)
    db.close()

    if 0 == len(user_li):
        return False

    if int(user_li[0]['level']) != user_admin:
        return False

    return True
コード例 #2
0
ファイル: controller.py プロジェクト: Tyler-D/Zfile
class BaseAction(object):
    resp = None
    template_dir = settings['TEMPLATE_DIR']
    theme_dir = settings['TEMPLATE_THEME_DIR']
    def __init__(self):
        self.db = MySQLDB_(settings['MYSQL_HOST'], settings['MYSQL_USER'],
                settings['MYSQL_PASSWD'], settings['MYSQL_DB'])

    def get(self):
        return web.ctx.method == 'GET'

    def req_params(self):
        return web.input()

    def session(self):
        return web.ctx.session

    def redirect(self, uri):
        '''301
        '''
        self.db.close()
        raise web.seeother(uri)

    def notfound(self):
        self.db.close()
        return web.notfound('404 Not Found.')

    def _print(self, page_name, base=None):
        """显示模板
        """
        if base:
            render = web.template.render(fileutil.join_path(self.template_dir, self.theme_dir),
                base=base)
        else:
            render = web.template.render(fileutil.join_path(self.template_dir, self.theme_dir))
        self.db.close()
        return getattr(render, page_name)(self.resp)

    def error(self):
        '''出错
        '''
        self.db.close()
        return self._print('error')

    def json(self):
        """response json
        从webpy返回json被chrome认为是错误的
        http://www.douban.com/group/topic/32487954/
        """
        self.db.close()
        return simplejson.dumps(self.resp)

    def xml(self, xml_name):
        """response xml
        提供XML访问
        http://webpy.org/cookbook/xmlfiles.zh-cn
        """
        web.header('Content-Type', 'text/xml')
        self.db.close()
        return self._print(xml_name)