first commit

Author: aodag

alembic.ini

@@ -0,0 +1,46 @@
+# A generic, single database configuration.
+
+[alembic]
+# path to migration scripts
+script_location = alembic
+
+# template used to generate migration files
+# file_template = %%(rev)s_%%(slug)s
+
+sqlalchemy.url = sqlite:///%(here)s/shirly.db
+
+
+# Logging configuration
+[loggers]
+keys = root,sqlalchemy,alembic
+
+[handlers]
+keys = console
+
+[formatters]
+keys = generic
+
+[logger_root]
+level = WARN
+handlers = console
+qualname =
+
+[logger_sqlalchemy]
+level = WARN
+handlers =
+qualname = sqlalchemy.engine
+
+[logger_alembic]
+level = INFO
+handlers =
+qualname = alembic
+
+[handler_console]
+class = StreamHandler
+args = (sys.stderr,)
+level = NOTSET
+formatter = generic
+
+[formatter_generic]
+format = %(levelname)-5.5s [%(name)s] %(message)s
+datefmt = %H:%M:%S

alembic/README

@@ -0,0 +1 @@
+Generic single-database configuration.

alembic/env.py

@@ -0,0 +1,72 @@
+from __future__ import with_statement
+from alembic import context
+from sqlalchemy import engine_from_config, pool
+from logging.config import fileConfig
+
+# this is the Alembic Config object, which provides
+# access to the values within the .ini file in use.
+config = context.config
+
+# Interpret the config file for Python logging. 
+# This line sets up loggers basically.
+fileConfig(config.config_file_name)
+
+# add your model's MetaData object here
+# for 'autogenerate' support
+# from myapp import mymodel
+# target_metadata = mymodel.Base.metadata
+import shirly.models
+target_metadata = shirly.models.Base.metadata
+
+# other values from the config, defined by the needs of env.py,
+# can be acquired:
+# my_important_option = config.get_main_option("my_important_option")
+# ... etc.
+
+def run_migrations_offline():
+    """Run migrations in 'offline' mode.
+
+    This configures the context with just a URL
+    and not an Engine, though an Engine is acceptable
+    here as well.  By skipping the Engine creation
+    we don't even need a DBAPI to be available.
+    
+    Calls to context.execute() here emit the given string to the
+    script output.
+    
+    """
+    url = config.get_main_option("sqlalchemy.url")
+    context.configure(url=url)
+
+    with context.begin_transaction():
+        context.run_migrations()
+
+def run_migrations_online():
+    """Run migrations in 'online' mode.
+
+    In this scenario we need to create an Engine
+    and associate a connection with the context.
+    
+    """
+    engine = engine_from_config(
+                config.get_section(config.config_ini_section), 
+                prefix='sqlalchemy.', 
+                poolclass=pool.NullPool)
+
+    connection = engine.connect()
+    context.configure(
+                connection=connection, 
+                target_metadata=target_metadata
+                )
+
+    try:
+        with context.begin_transaction():
+            context.run_migrations()
+    finally:
+        connection.close()
+
+if context.is_offline_mode():
+    run_migrations_offline()
+else:
+    run_migrations_online()
+

alembic/script.py.mako

@@ -0,0 +1,22 @@
+"""${message}
+
+Revision ID: ${up_revision}
+Revises: ${down_revision}
+Create Date: ${create_date}
+
+"""
+
+# revision identifiers, used by Alembic.
+revision = ${repr(up_revision)}
+down_revision = ${repr(down_revision)}
+
+from alembic import op
+import sqlalchemy as sa
+${imports if imports else ""}
+
+def upgrade():
+    ${upgrades if upgrades else "pass"}
+
+
+def downgrade():
+    ${downgrades if downgrades else "pass"}

alembic/versions/4d46b1bd74d7_.py

@@ -0,0 +1,31 @@
+"""empty message
+
+Revision ID: 4d46b1bd74d7
+Revises: None
+Create Date: 2012-04-13 08:14:39.706046
+
+"""
+
+# revision identifiers, used by Alembic.
+revision = '4d46b1bd74d7'
+down_revision = None
+
+from alembic import op
+import sqlalchemy as sa
+
+
+def upgrade():
+    ### commands auto generated by Alembic - please adjust! ###
+    op.create_table('users',
+    sa.Column('id', sa.Integer(), nullable=False),
+    sa.Column('user_name', sa.Unicode(length=255), nullable=True, unique=True),
+    sa.Column('password', sa.String(length=255), nullable=True),
+    sa.PrimaryKeyConstraint('id')
+    )
+    ### end Alembic commands ###
+
+
+def downgrade():
+    ### commands auto generated by Alembic - please adjust! ###
+    op.drop_table('users')
+    ### end Alembic commands ###

development.ini

@@ -0,0 +1,45 @@
+[app:main]
+paste.app_factory = shirly:main
+pyramid.debug_authentication = true
+pyramid.debug_authorization = true
+pyramid.reload_templates = true
+pyramid.includes = 
+    pyramid_tm
+    shirly.security
+who.ini = %(here)s/who.ini
+sqlalchemy.url = sqlite:///%(here)s/shirly.db
+sqlalchemy.echo = true
+
+[server:main]
+use = egg:waitress
+host = 0.0.0.0
+port = 6543
+
+[loggers]
+keys=root,simpleExample
+
+[handlers]
+keys=consoleHandler
+
+[formatters]
+keys=simpleFormatter
+
+[logger_root]
+level=DEBUG
+handlers=consoleHandler
+
+[logger_simpleExample]
+level=DEBUG
+handlers=consoleHandler
+qualname=simpleExample
+propagate=0
+
+[handler_consoleHandler]
+class=StreamHandler
+level=DEBUG
+formatter=simpleFormatter
+args=(sys.stdout,)
+
+[formatter_simpleFormatter]
+format=%(asctime)s - %(name)s - %(levelname)s - %(message)s
+datefmt=

setup.cfg

@@ -0,0 +1,2 @@
+[aliases]
+dev = develop easy_install shirly[testing] shirly[docs] shirly[dev]

setup.py

@@ -0,0 +1,44 @@
+from setuptools import setup, find_packages
+
+requires = [
+    "pyramid",
+    "pyramid-tm",
+    "pyramid-who",
+    "pyramid-fanstatic",
+    "pyramid-simpleform",
+    "sqlahelper",
+    "repoze.who",
+    "repoze.who.plugins.sa",
+    "waitress",
+    "js.jquery",
+    "js.jqueryui",
+    "js.tinymce",
+]
+
+tests_require = [
+    "nose",
+    "webtest",
+    "coverage",
+    "tox",
+    "mock",
+]
+
+points = {
+    'console_scripts': [
+        'add_user=shirly.scripts:add_user',
+    ],
+}
+
+setup(
+    name="shirly",
+    install_requires=requires,
+    tests_require=tests_require,
+    extras_require={
+        "testing": tests_require,
+        "docs": ["sphinx"],
+        "dev": ["alembic"],
+    },
+    test_suite="shirly",
+    package_dir={"": "src"},
+    entry_points=points,
+)

src/shirly/__init__.py

@@ -0,0 +1,15 @@
+#
+from pyramid.config import Configurator
+import sqlahelper
+from sqlalchemy import engine_from_config
+
+def main(global_config, **settings):
+    engine = engine_from_config(settings)
+    sqlahelper.add_engine(engine)
+    config = Configurator(settings=settings,
+        root_factory='.models.ShirlyResource')
+    config.add_route('top', '/')
+    config.add_route('login', '/login')
+    config.add_route('logout', '/logout')
+    config.scan()
+    return config.make_wsgi_app()

src/shirly/models.py

@@ -0,0 +1,30 @@
+import hashlib
+import sqlahelper
+import sqlalchemy as sa
+from pyramid.security import Everyone, Allow, Authenticated
+
+class ShirlyResource(object):
+    __acl__ = [
+        (Allow, Authenticated, 'viewer'),
+    ]
+    def __init__(self, request):
+        self.request = request
+
+
+Base = sqlahelper.get_base()
+DBSession = sqlahelper.get_session()
+
+class User(Base):
+    __tablename__ = 'users'
+
+    id = sa.Column(sa.Integer, primary_key=True)
+    user_name = sa.Column(sa.Unicode(255), unique=True)
+    _password = sa.Column("password", sa.String(255))
+
+    def set_password(self, password):
+        self._password = hashlib.sha1(password).hexdigest()
+
+    password = property(fset=set_password)
+
+    def validate_password(self, password):
+        return self._password == hashlib.sha1(password).hexdigest()

src/shirly/scripts.py

@@ -0,0 +1,15 @@
+import sys
+from pyramid.paster import bootstrap
+import sqlahelper
+from . import models as m
+
+def add_user():
+    env = bootstrap(sys.argv[1])
+    settings = env['registry'].settings
+
+    user_name = raw_input('user name: ')
+    password = raw_input('password: ')
+    user = m.User(user_name=user_name, password=password)
+    m.DBSession.add(user)
+    import transaction
+    transaction.commit()

src/shirly/security.py

@@ -0,0 +1,36 @@
+import logging
+from pyramid.authorization import ACLAuthorizationPolicy
+from pyramid_who.whov2 import WhoV2AuthenticationPolicy
+from repoze.who.api import get_api as get_who_api
+from repoze.who.interfaces import IAPIFactory
+
+def includeme(config):
+    who_ini = config.registry.settings['who.ini']
+    authorization_policy = ACLAuthorizationPolicy()
+    authentication_policy = WhoV2AuthenticationPolicy(who_ini,
+        identifier_id='auth_tkt')
+    api_factory = authentication_policy._api_factory
+    config.registry.registerUtility(api_factory, IAPIFactory)
+    config.set_authorization_policy(authorization_policy)
+    config.set_authentication_policy(authentication_policy)
+
+    config.add_tween('shirly.security.tween_factory')
+
+    config.set_request_property(authenticated_user, 'authenticated_user')
+
+def tween_factory(handler, registry):
+    api_factory = registry.getUtility(IAPIFactory)
+    def who_api_tween(request):
+        api_factory(request.environ)
+        return handler(request)
+    return who_api_tween
+
+def authenticate(request):
+    who_api = get_who_api(request.environ)
+    identity, headers = who_api.login({'login': request.params['login'], 'password': request.params['password']})
+    logging.debug('logged in %s' % identity)
+    logging.debug('headers in %s' % headers)
+    return identity.get('repoze.who.userid')
+
+def authenticated_user(request):
+    return request.environ.get('repoze.who.identity', {}).get('user')

src/shirly/templates/index.mak

@@ -0,0 +1,3 @@
+<h1>Shirly</h1>
+${request.authenticated_user.user_name}
+<a href="${request.route_url('logout')}">Logout</a>

src/shirly/templates/login.mak

@@ -0,0 +1,6 @@
+<h1>Login</h1>
+<form method="POST" action="${request.route_url('login')}" >
+<input type="text" name="login" />
+<input type="password" name="password" />
+<input type="submit" value="Login" />
+</form>