示例#1
0
def setup_mysql_db(request):

    if os.environ.get("TRAVIS", False):
        # Using Travis-CI - https://travis-ci.org/
        password = ""
        db_name = "myapp_test"
    else:
        password = ""
        db_name = "cork_functional_test"

    uri = "mysql://*****:*****@localhost/%s" % (password, db_name)
    mb = SqlAlchemyBackend(uri, initialize=True)

    ## Purge DB
    mb._drop_all_tables()

    assert len(mb.roles) == 0
    assert len(mb.users) == 0

    # Create roles
    mb.roles.insert({"role": "special", "level": 200})
    mb.roles.insert({"role": "admin", "level": 100})
    mb.roles.insert({"role": "editor", "level": 60})
    mb.roles.insert({"role": "user", "level": 50})

    # Create admin
    mb.users.insert(
        {
            "username": "******",
            "email_addr": "*****@*****.**",
            "desc": "admin test user",
            "role": "admin",
            "hash": "cLzRnzbEwehP6ZzTREh3A4MXJyNo+TV8Hs4//EEbPbiDoo+dmNg22f2RJC282aSwgyWv/O6s3h42qrA6iHx8yfw=",
            "creation_date": "2012-10-28 20:50:26.286723",
            "last_login": "******",
        }
    )
    assert len(mb.roles) == 4
    assert len(mb.users) == 1

    def fin():
        return  # TODO: fix
        mb._drop_all_tables()
        assert len(mb.roles) == 0
        assert len(mb.users) == 0

    request.addfinalizer(fin)
    return mb
示例#2
0
def setup_mysql_db(request):

    if os.environ.get('TRAVIS', False):
        # Using Travis-CI - https://travis-ci.org/
        password = ''
        db_name = 'myapp_test'
    else:
        password = ''
        db_name = 'cork_functional_test'

    uri = "mysql://*****:*****@localhost/%s" % (password, db_name)
    mb = SqlAlchemyBackend(uri, initialize=True)

    ## Purge DB
    mb._drop_all_tables()

    assert len(mb.roles) == 0
    assert len(mb.users) == 0

    # Create roles
    mb.roles.insert({'role': 'special', 'level': 200})
    mb.roles.insert({'role': 'admin', 'level': 100})
    mb.roles.insert({'role': 'editor', 'level': 60})
    mb.roles.insert({'role': 'user', 'level': 50})

    # Create admin
    mb.users.insert({
        "username": "******",
        "email_addr": "*****@*****.**",
        "desc": "admin test user",
        "role": "admin",
        "hash": "cLzRnzbEwehP6ZzTREh3A4MXJyNo+TV8Hs4//EEbPbiDoo+dmNg22f2RJC282aSwgyWv/O6s3h42qrA6iHx8yfw=",
        "creation_date": "2012-10-28 20:50:26.286723",
        "last_login": "******"
    })
    assert len(mb.roles) == 4
    assert len(mb.users) == 1

    def fin():
        return  # TODO: fix
        mb._drop_all_tables()
        assert len(mb.roles) == 0
        assert len(mb.users) == 0

    request.addfinalizer(fin)
    return mb
示例#3
0
def setup(engine):
    Role.metadata.create_all(engine)
    User.metadata.create_all(engine)

    # setup cork auth
    backend = SqlAlchemyBackend(engine.url)
    aaa = Cork(backend=backend)
    setattr(__import__(__name__), 'aaa', aaa)
示例#4
0
def setup_postgresql_db(request):

    if os.environ.get('TRAVIS', False):
        # Using Travis-CI - https://travis-ci.org/
        db_name = 'myapp_test'
    else:
        db_name = 'cork_functional_test'

    uri = "postgresql+psycopg2://postgres:@/%s" % db_name
    mb = SqlAlchemyBackend(uri, initialize=True)

    # Purge DB
    mb._drop_all_tables()
    assert len(mb.roles) == 0
    assert len(mb.users) == 0

    # Create roles
    mb.roles.insert({'role': 'special', 'level': 200})
    mb.roles.insert({'role': 'admin', 'level': 100})
    mb.roles.insert({'role': 'editor', 'level': 60})
    mb.roles.insert({'role': 'user', 'level': 50})

    # Create admin
    mb.users.insert({
        "username": "******",
        "email_addr": "*****@*****.**",
        "desc": "admin test user",
        "role": "admin",
        "hash":
        "cLzRnzbEwehP6ZzTREh3A4MXJyNo+TV8Hs4//EEbPbiDoo+dmNg22f2RJC282aSwgyWv/O6s3h42qrA6iHx8yfw=",
        "creation_date": "2012-10-28 20:50:26.286723",
        "last_login": "******"
    })
    assert len(mb.roles) == 4
    assert len(mb.users) == 1

    def fin():
        mb._drop_all_tables()
        assert len(mb.roles) == 0
        assert len(mb.users) == 0

    request.addfinalizer(fin)
    return mb
示例#5
0
def change_pass(pwd):
    sqlalchemy_backend = Cork(backend=SqlAlchemyBackend(config.db_url))
    metadata = MetaData()
    engine = create_engine(config.db_url)
    metadata.bind = engine
    users = Table("users",
                  metadata,
                  extend_existing=True,
                  autoload=True,
                  autoload_with=engine)
    users.update().where(users.c.username == "transito").values(
        hash=sqlalchemy_backend._hash("transito", pwd)).execute()
示例#6
0
def connect_to_test_db():

    if os.environ.get('TRAVIS', False):
        # Using Travis-CI - https://travis-ci.org/
        password = ''
        db_name = 'myapp_test'
    else:
        password = ''
        db_name = 'cork_functional_test'

    uri = "mysql://*****:*****@localhost/%s" % (password, db_name)
    return SqlAlchemyBackend(uri, initialize=True)
def add_user(Base, session):
    Users = Base.classes.users
    sqlalchemy_backend = Cork(backend=SqlAlchemyBackend(config.db_url))
    session.add(
        Users(username="******",
              role="admin",
              hash=sqlalchemy_backend._hash("transito", "transito"),
              email_addr="*****@*****.**",
              desc="operario",
              creation_date=str(datetime.now()),
              last_login=str(datetime.now())))
    session.commit()
    session.close()
示例#8
0
def setup_sqlalchemy_with_sqlite_in_memory_db(request):
    # Setup an SqlAlchemyBackend backend using an in-memory SQLite DB

    mb = SqlAlchemyBackend("sqlite:///:memory:", initialize=True)

    ## Purge DB
    mb._drop_all_tables()
    assert len(mb.roles) == 0
    assert len(mb.users) == 0

    # Create roles
    mb.roles.insert({"role": "special", "level": 200})
    mb.roles.insert({"role": "admin", "level": 100})
    mb.roles.insert({"role": "editor", "level": 60})
    mb.roles.insert({"role": "user", "level": 50})

    # Create admin
    mb.users.insert(
        {
            "username": "******",
            "email_addr": "*****@*****.**",
            "desc": "admin test user",
            "role": "admin",
            "hash": "cLzRnzbEwehP6ZzTREh3A4MXJyNo+TV8Hs4//EEbPbiDoo+dmNg22f2RJC282aSwgyWv/O6s3h42qrA6iHx8yfw=",
            "creation_date": "2012-10-28 20:50:26.286723",
            "last_login": "******",
        }
    )
    assert len(mb.roles) == 4
    assert len(mb.users) == 1

    def fin():
        mb._drop_all_tables()
        assert len(mb.roles) == 0
        assert len(mb.users) == 0

    request.addfinalizer(fin)
    return mb
示例#9
0
def setup_sqlalchemy_with_sqlite_in_memory_db(request):
    # Setup an SqlAlchemyBackend backend using an in-memory SQLite DB

    mb = SqlAlchemyBackend('sqlite:///:memory:', initialize=True)

    ## Purge DB
    mb._drop_all_tables()
    assert len(mb.roles) == 0
    assert len(mb.users) == 0

    # Create roles
    mb.roles.insert({'role': 'special', 'level': 200})
    mb.roles.insert({'role': 'admin', 'level': 100})
    mb.roles.insert({'role': 'editor', 'level': 60})
    mb.roles.insert({'role': 'user', 'level': 50})

    # Create admin
    mb.users.insert({
        "username": "******",
        "email_addr": "*****@*****.**",
        "desc": "admin test user",
        "role": "admin",
        "hash": "cLzRnzbEwehP6ZzTREh3A4MXJyNo+TV8Hs4//EEbPbiDoo+dmNg22f2RJC282aSwgyWv/O6s3h42qrA6iHx8yfw=",
        "creation_date": "2012-10-28 20:50:26.286723",
        "last_login": "******"
    })
    assert len(mb.roles) == 4
    assert len(mb.users) == 1

    def fin():
        mb._drop_all_tables()
        assert len(mb.roles) == 0
        assert len(mb.users) == 0

    request.addfinalizer(fin)
    return mb
示例#10
0
def auth_sqlalchemy():
    sqlalchemy_backend = SqlAlchemyBackend(config.db_url)
    return sqlalchemy_backend
示例#11
0
import logging
import smtplib
from operator import itemgetter
import os

MODULEPATH = os.path.dirname(__file__)
TEMPLATE_PATH.insert(0, os.path.join(MODULEPATH, "views"))

logging.basicConfig(format='localhost - - [%(asctime)s] %(message)s',
                    level=logging.DEBUG)
log = logging.getLogger(__name__)

postgresConn = os.environ.get("POSTGRES_CONN", "")
sessionEncryptKey = os.environ.get("SESSION_KEY", "asdfghjkl")
devMode = os.environ.get("DEV_MODE", "disabled")
corkBackend = SqlAlchemyBackend(postgresConn, initialize=False)
aaa = Cork(backend=corkBackend, email_sender='', smtp_url='')


# Section: Utilities


def postd():
    return bottle.request.forms


def post_get(name, default=''):
    return bottle.request.POST.get(name, default).strip()


def get_get(name, default=''):
def purge_test_db():
    # Purge DB
    mb = SqlAlchemyBackend('mysql://*****:*****@localhost/cork_functional_test', initialize=True)
    mb._drop_all_tables()
示例#13
0
def getMysqlBackend():
    connectionString = config.generateConnectioString()
    sqlBackend = SqlAlchemyBackend(db_full_url=connectionString)
    return sqlBackend
示例#14
0
文件: auth.py 项目: deb17/regex
COOKIE_SECRET = os.environ['COOKIE_SECRET']

# SITE_VERIFY_URL = 'https://www.google.com/recaptcha/api/siteverify'
# RECAPTCHA_SECRET = '6LfeHx4UAAAAAFWXGh_xcL0B8vVcXnhn9q_SnQ1b'

logging.basicConfig(format='heroku - - [%(asctime)s] %(message)s',
                    level=logging.DEBUG)

from cork.backends import SqlAlchemyBackend

if not DATABASE_URL.startswith('postgresql'):
    DATABASE_URL = 'postgresql:' + DATABASE_URL.split(':', 1)[1]

sa = SqlAlchemyBackend(DATABASE_URL,
                       initialize=True,
                       connect_args={'dbname': DBNAME})

SMTP_URL = ('starttls://' + MAILID + ':' + PASSWORD +
            '@smtp-mail.outlook.com:587')
aaa = Cork(smtp_url=SMTP_URL, email_sender=MAILID, backend=sa)

authorize = aaa.make_auth_decorator(fail_redirect="/login", role="user")

app = bottle.Bottle()
app.install(FlashPlugin(secret=COOKIE_SECRET))


def postd():
    return bottle.request.forms