def listuserinfo():
arguments = listuserparser.parse_args()
if not arguments.config or not arguments.username:
listuserparser.print_usage()
else:
config_uri = arguments.config
setup_logging(config_uri)
try:
settings = get_appsettings(config_uri, 'factored')
except LookupError:
settings = get_appsettings(config_uri, 'main')
engine = engine_from_config(settings, 'sqlalchemy.')
DBSession.configure(bind=engine)
session = DBSession()
users = session.query(User).filter_by(
username=arguments.username).all()
if len(users) > 0:
user = users[0]
print('username:%s, secret: %s' % (user.username, user.secret))
print(
'bar code url:',
get_barcode_image(user.username, user.secret,
settings['appname']))
else:
print('"%s" user not found' % arguments.username)
def listusers():
arguments = removeparser.parse_args()
if not arguments.config:
removeparser.print_usage()
else:
config_uri = arguments.config
setup_logging(config_uri)
try:
settings = get_appsettings(config_uri, 'factored')
except LookupError:
settings = get_appsettings(config_uri, 'main')
engine = engine_from_config(settings, 'sqlalchemy.')
DBSession.configure(bind=engine)
session = DBSession()
for user in session.query(User).all():
print(user.username)
def remove():
arguments = removeparser.parse_args()
if not arguments.config or not arguments.username:
removeparser.print_usage()
else:
config_uri = arguments.config
setup_logging(config_uri)
try:
settings = get_appsettings(config_uri, 'factored')
except LookupError:
settings = get_appsettings(config_uri, 'main')
engine = engine_from_config(settings, 'sqlalchemy.')
DBSession.configure(bind=engine)
session = DBSession()
user = session.query(User).filter_by(username=arguments.username).all()
if len(user) > 0:
session.delete(user[0])
else:
print('"%s" user not found' % arguments.username)
session.commit()
session.close()
def add():
arguments = addparser.parse_args()
if not arguments.config or not arguments.username:
addparser.print_usage()
else:
config_uri = arguments.config
setup_logging(config_uri)
try:
settings = get_appsettings(config_uri, 'factored')
except LookupError:
settings = get_appsettings(config_uri, 'main')
engine = engine_from_config(settings, 'sqlalchemy.')
DBSession.configure(bind=engine)
session = DBSession()
username = arguments.username
user = create_user(username, session)
print('barcode url:',
get_barcode_image(username, user.secret, settings['appname']))
print('secret:', user.secret)
session.commit()
session.close()
def setUp(self):
self.config = testing.setUp()
self.settings = {
'sqlalchemy.url': 'sqlite://',
'em.subject': 'Authentication Request',
'em.sender': '*****@*****.**',
'em.body': """
You have requested authentication.
Your temporary access code is: {code}""",
'auth_tkt.secret': 'secret',
'auth_tkt.cookie_name': 'pnutbtr',
'supported_auth_schemes': ['Google Auth', 'Email']
}
self.app = Authenticator(test_application, {}, **self.settings)
from webtest import TestApp
self.testapp = TestApp(self.app)
self.mailer = FakeMailer()
self.testapp.app.pyramid.registry['mailer'] = self.mailer
from sqlalchemy import create_engine
engine = create_engine('sqlite://')
from factored.models import Base
DBSession.configure(bind=engine)
self.session = DBSession()
Base.metadata.create_all(engine)
