Example #1
0
def tag_session(req):
    app = get_app()
    cookie_name = app.conf['sessions/cookie_name']
    session = SecureCookie.load_cookie(req, cookie_name,
                                           app.conf['sessions/secret'])
    req.session = session
    local.session = session
Example #2
0
def test_check_login():
    from glashammer.bundles.auth import setup_auth, login
    from glashammer.bundles.sessions import get_session

    called = []

    def check(u, p):
        called.append((u, p))
        return u == p

    def view(req):
        login('a')
        return Response()

    def setup_app(app):
        app.add_setup(setup_auth)
        app.connect_event('password-check', check)
        app.add_url('/', 'a/b', view=view)

    app = make_app(setup_app, 'test_output')
    c = Client(app)
    c.open()

    session = get_session()
    token_key = get_app().conf['auth/token_key']
    assert token_key in session
    assert session[token_key] == 'a'
Example #3
0
def _date_format(formatter, obj, format):
    app = get_app()
    if app is None:
        locale = Locale('en')
    else:
        locale = app.locale
    return formatter(obj, format, locale=locale)
Example #4
0
def get_engine():
    """Get the SQLAlchemy DB Engine

    This returns the global SQLALchemy engine, which is *usually* shared amongst all
    threads.
    """
    return get_app().sqla_db_engine
Example #5
0
def ngettext(singular, plural, n):
    """Translate the possible pluralized string to the language of the
    application.
    """
    app = get_app()
    if app is None:
        if n == 1:
            return singular
        return plural
    return app.translations.ungettext(singular, plural, n)
Example #6
0
def cleanup_sessions(response):
    # save the session
    app = get_app()
    session = get_session()
    if session.should_save:
        cookie_name = app.conf['sessions/cookie_name']
        if session.get('pmt'):
            max_age = 60 * 60 * 24 * 31
            expires = time() + max_age
        else:
            max_age = expires = None
        session.save_cookie(response, cookie_name, max_age=max_age,
                            expires=expires, session_expires=expires)
Example #7
0
def parse_datetime(string):
    """Parses a string into a datetime object."""
    if string.lower() == _('now'):
        return datetime.utcnow()
    convert = lambda fmt: datetime(*strptime(string, fmt)[:7])
    cfg = get_app().cfg

    # first of all try the following format because a while ago it was
    # Textpress' default format
    try:
        return convert(u'%Y-%m-%d %H:%M')
    except ValueError:
        pass

    # no go with time only, and current day
    base = datetime.utcnow()
    for fmt in TIME_FORMATS:
        try:
            val = convert(fmt)
        except ValueError:
            continue
        return base.replace(hour=val.hour, minute=val.minute,
                            second=val.second)

    # no try date + time
    def combined():
        for t_fmt in TIME_FORMATS:
            for d_fmt in DATE_FORMATS:
                yield t_fmt + ' ' + d_fmt
                yield d_fmt + ' ' + t_fmt

    for fmt in combined():
        try:
            return convert(fmt)
        except ValueError:
            pass

    raise ValueError('invalid date format')
Example #8
0
def list_languages():
    """Return a list of all languages."""
    app = get_app()
    if app:
        locale = app.locale
    else:
        locale = Locale('en')

    languages = [('en', Locale('en').get_display_name(locale))]
    folder = os.path.dirname(__file__)

    for filename in os.listdir(folder):
        if filename == 'en' or not \
           os.path.isdir(os.path.join(folder, filename)):
            continue
        try:
            l = Locale.parse(filename)
        except UnknownLocaleError:
            continue
        languages.append((str(l), l.get_display_name(locale)))

    languages.sort(key=lambda x: x[1].lower())
    return languages
Example #9
0
def get_cache():
    return get_app().cache
Example #10
0
 def test_engine(self):
     assert self.app.sqla_db_engine is get_app().sqla_db_engine
Example #11
0
 def test_engine(self):
     assert get_app().sqla_db_engine
Example #12
0
 def _bview(req):
     app = get_app()
     assert_raises((RuntimeError,), app.add_config_var, 'a', str, 'a')
     return Response('hello')
Example #13
0
 def test_get_app(self):
     assert get_app() is self.app
Example #14
0
def logout():
    session = get_session()
    app = get_app()
    del session[app.conf['auth/token_key']]
Example #15
0
def login(token):
    session = get_session()
    app = get_app()
    session[app.conf['auth/token_key']] = token
Example #16
0
def get_username():
    session = get_session()
    app = get_app()
    return session.get(app.conf['auth/token_key'])
Example #17
0
def get_engine():
    """Get the SQLA DB Engine"""
    return get_app().sqla_db_engine
Example #18
0
def init_data(app):
    engine = get_app().sqla_db_engine
    metadata.create_all(engine)
Example #19
0
def _generate_token():
    return sha1("%s#%s" % (time(), get_app().cfg['sessions/secret'])).hexdigest()
Example #20
0
 def _open_conn():
     db = get_app().zodb_db
     local.zodb_db = db
     local.zodb_conn = db.open()
Example #21
0
def gettext(string):
    """Translate the given string to the language of the application."""
    app = get_app()
    if app is None:
        return string
    return app.translations.ugettext(string)
Example #22
0
def initialize_manager(*args):
    """
    Ensure there is a thread local isntance of the stores manager.
    """
    if not hasattr(local, 'storm_stores_manager'):
        local.storm_stores_manager = StoreManager(get_app().conf['stormdb/dburi'])
Example #23
0
 def wrapped(*args, **kw):
     if get_app().conf['auth/token_key'] in get_session():
         return f(*args, **kw)
     else:
         return redirect(url_for('auth/login'))