def auth_meta_mw(resp):
"""
Callback for meta.wikimedia.org to send us authentication results.
This is responsible for fetching existing users or creating new ones.
If a new user is created, they get the default role of GUEST and
an email or username to match their details from the OAuth provider.
"""
if resp is None:
flash('You need to grant the app permissions in order to login.', 'error')
return redirect(url_for('login'))
session['access_token'] = (
resp['oauth_token'],
resp['oauth_token_secret']
)
try:
identify_token_encoded = meta_mw.post(
app.config['META_MW_BASE_URL'] + app.config['META_MW_IDENTIFY_URI'],
).data
identify_token = process_mw_jwt(identify_token_encoded)
username = identify_token['username']
userid = identify_token['sub']
db_session = db.get_session()
user = None
try:
user = db_session.query(User).filter_by(meta_mw_id=userid).one()
except NoResultFound:
user = User(
username=username,
meta_mw_id=userid,
role=UserRole.GUEST,
)
db_session.add(user)
db_session.commit()
except MultipleResultsFound:
db_session.close()
return 'Multiple users found with your id!!! Contact Administrator'
user.login(db_session)
try:
if login_user(user):
user.detach_from(db_session)
redirect_to = session.get('next') or url_for('home_index')
redirect_to = urllib2.unquote(redirect_to)
return redirect(redirect_to)
finally:
db_session.close()
except Exception, e:
flash('Access to this application was revoked. Please re-login!')
app.logger.exception(str(e))
return redirect(url_for('login'))
def auth_google(resp):
"""
Callback for Google to send us authentication results.
This is responsible for fetching existing users or creating new ones.
If a new user is created, they get the default role of GUEST and
an email or username to match their details from the OAuth provider.
"""
if resp is None and request.args.get('error') == 'access_denied':
flash('You need to grant the app permissions in order to login.', 'error')
return redirect(url_for('login'))
access_token = resp['access_token'] or request.args.get('code')
if access_token:
session['access_token'] = access_token, ''
r = requests.get(app.config['GOOGLE_USERINFO_URI'], headers={
'Authorization': 'OAuth ' + access_token
})
if r.ok:
userinfo = json.loads(r.text)
email = userinfo['email']
id = userinfo['id']
db_session = db.get_session()
user = None
try:
user = db_session.query(User).filter_by(google_id=id).one()
except NoResultFound:
user = User(
email=email,
google_id=id,
role=UserRole.GUEST,
)
db_session.add(user)
db_session.commit()
except MultipleResultsFound:
db_session.close()
return 'Multiple users found with your id!!! Contact Administrator'
try:
user.login(db_session)
if login_user(user):
user.detach_from(db_session)
redirect_to = session.get('next') or url_for('home_index')
redirect_to = urllib2.unquote(redirect_to)
return redirect(redirect_to)
finally:
db_session.close()
flash('Was not allowed to authenticate you with Google.', 'error')
return redirect(url_for('login'))
def load_user(user_id):
"""
Callback required by Flask-Login. Gets the User object from the database.
"""
db_session = db.get_session()
try:
user = User.get(db_session, user_id)
finally:
db_session.close()
return user
def setUp(self):
#****************************************************************
# set up for every test - delete and re-create all needed records
#****************************************************************
project = 'enwiki'
self.session = db.get_session()
engine = db.get_mw_engine(project)
db.MediawikiBase.metadata.create_all(engine, checkfirst=True)
self.mwSession = db.get_mw_session(project)
DatabaseTest.tearDown(self)
#****************************************************************
# create records for enwiki tests
#****************************************************************
mw_user_dan = MediawikiUser(user_name='Dan')
mw_user_evan = MediawikiUser(user_name='Evan')
mw_user_andrew = MediawikiUser(user_name='Andrew')
mw_user_diederik = MediawikiUser(user_name='Diederik')
mw_logging = Logging(log_user_text='Reedy')
mw_page = Page(page_namespace=0, page_title='Main_Page')
mw_second_page = Page(page_namespace=209, page_title='Page in Namespace 209')
self.mwSession.add_all([
mw_user_dan,
mw_user_evan,
mw_user_andrew,
mw_logging,
mw_page,
mw_second_page,
])
self.mwSession.commit()
self.createTestDataMetricPagesCreated(mw_user_evan)
# edits in between Dan and Evan edits
rev_before_1 = Revision(
rev_page=mw_page.page_id, rev_user=mw_user_diederik.user_id,
rev_comment='before Dan edit 1',
rev_len=4, rev_timestamp=datetime(2013, 05, 30),
)
rev_before_2 = Revision(
rev_page=mw_page.page_id, rev_user=mw_user_diederik.user_id,
rev_comment='before Dan edit 2',
rev_len=0, rev_timestamp=datetime(2013, 06, 30),
)
rev_before_3 = Revision(
rev_page=mw_page.page_id, rev_user=mw_user_diederik.user_id,
rev_comment='before Evan edit 1',
rev_len=0, rev_timestamp=datetime(2013, 05, 30),
)
rev_before_4 = Revision(
rev_page=mw_page.page_id, rev_user=mw_user_diederik.user_id,
rev_comment='before Evan edit 2',
rev_len=100, rev_timestamp=datetime(2013, 06, 30),
)
rev_before_5 = Revision(
rev_page=mw_page.page_id, rev_user=mw_user_diederik.user_id,
rev_comment='before Evan edit 3',
rev_len=140, rev_timestamp=datetime(2013, 07, 23),
)
rev_alternate_namespace_1 = Revision(
rev_page=mw_second_page.page_id, rev_user=mw_user_dan.user_id,
rev_comment='first revision in namespace 209',
# NOTE: VIM is freaking out if I type 08 below. Is this true on Mac?
rev_len=100, rev_timestamp=datetime(2013, 8, 5),
)
self.mwSession.add_all([
rev_before_1,
rev_before_2,
rev_before_3,
rev_before_4,
rev_before_5,
rev_alternate_namespace_1,
])
self.mwSession.commit()
# Dan edits
rev1 = Revision(
rev_page=mw_page.page_id, rev_user=mw_user_dan.user_id,
rev_comment='Dan edit 1', rev_parent_id=rev_before_1.rev_id,
rev_len=0, rev_timestamp=datetime(2013, 06, 01),
)
rev2 = Revision(
rev_page=mw_page.page_id, rev_user=mw_user_dan.user_id,
rev_comment='Dan edit 2', rev_parent_id=rev_before_2.rev_id,
rev_len=10, rev_timestamp=datetime(2013, 07, 01),
)
# Evan edits
rev3 = Revision(
rev_page=mw_page.page_id, rev_user=mw_user_evan.user_id,
rev_comment='Evan edit 1', rev_parent_id=rev_before_3.rev_id,
rev_len=100, rev_timestamp=datetime(2013, 06, 01),
)
rev4 = Revision(
rev_page=mw_page.page_id, rev_user=mw_user_evan.user_id,
rev_comment='Evan edit 2', rev_parent_id=rev_before_4.rev_id,
rev_len=140, rev_timestamp=datetime(2013, 07, 01),
)
rev5 = Revision(
rev_page=mw_page.page_id, rev_user=mw_user_evan.user_id,
rev_comment='Evan edit 3', rev_parent_id=rev_before_5.rev_id,
rev_len=136, rev_timestamp=datetime(2013, 07, 24),
)
self.mwSession.add_all([rev1, rev2, rev3, rev4, rev5])
self.mwSession.commit()
#****************************************************************
# create basic test records for non-mediawiki tests
#****************************************************************
dan_user = User(username='******')
evan_user = User(username='******')
web_test_user = User(email='*****@*****.**')
# create a test cohort
dan = WikiUser(
mediawiki_username=mw_user_dan.user_name,
mediawiki_userid=mw_user_dan.user_id,
project=project
)
evan = WikiUser(
mediawiki_username=mw_user_evan.user_name,
mediawiki_userid=mw_user_evan.user_id,
project=project
)
andrew = WikiUser(
mediawiki_username=mw_user_andrew.user_name,
mediawiki_userid=mw_user_andrew.user_id,
project=project
)
diederik = WikiUser(
mediawiki_username=mw_user_diederik.user_name,
mediawiki_userid=mw_user_diederik.user_id,
project=project
)
# create cohorts
test_cohort = Cohort(name='test', enabled=True, public=False)
private_cohort = Cohort(name='test_private', enabled=True, public=False)
private_cohort2 = Cohort(name='test_private2', enabled=True, public=False)
disabled_cohort = Cohort(name='test_disabled', enabled=False, public=False)
self.session.add_all([
#report,
dan_user,
evan_user,
web_test_user,
dan,
evan,
andrew,
diederik,
test_cohort,
private_cohort,
private_cohort2,
disabled_cohort])
self.session.commit()
# create cohort membership
dan_in_test = CohortWikiUser(wiki_user_id=dan.id, cohort_id=test_cohort.id)
evan_in_test = CohortWikiUser(wiki_user_id=evan.id, cohort_id=test_cohort.id)
andrew_in_test = CohortWikiUser(wiki_user_id=andrew.id, cohort_id=test_cohort.id)
diederik_in_test = CohortWikiUser(
wiki_user_id=diederik.id,
cohort_id=test_cohort.id
)
self.session.add_all([
dan_in_test,
evan_in_test,
andrew_in_test,
diederik_in_test
])
self.session.commit()
# create cohort ownership
dan_owns_test = CohortUser(
user_id=dan_user.id,
cohort_id=test_cohort.id,
role=CohortUserRole.OWNER,
)
evan_owns_private = CohortUser(
user_id=evan_user.id,
cohort_id=private_cohort.id,
role=CohortUserRole.OWNER,
)
evan_owns_private2 = CohortUser(
user_id=evan_user.id,
cohort_id=private_cohort2.id,
role=CohortUserRole.OWNER,
)
web_user_owns_test = CohortUser(
user_id=web_test_user.id,
cohort_id=test_cohort.id,
role=CohortUserRole.OWNER,
)
web_user_owns_private = CohortUser(
user_id=web_test_user.id,
cohort_id=private_cohort.id,
role=CohortUserRole.OWNER,
)
web_user_owns_private2 = CohortUser(
user_id=web_test_user.id,
cohort_id=private_cohort2.id,
role=CohortUserRole.OWNER,
)
dan_views_private2 = CohortUser(
user_id=dan_user.id,
cohort_id=private_cohort2.id,
role=CohortUserRole.VIEWER
)
self.session.add_all([
dan_owns_test,
evan_owns_private,
evan_owns_private2,
web_user_owns_test,
web_user_owns_private,
web_user_owns_private2,
dan_views_private2
])
self.session.commit()
# add reports
report_created = PersistentReport(
user_id=web_test_user.id,
status=celery.states.PENDING,
queue_result_key=None,
show_in_ui=True
)
report_started = PersistentReport(
user_id=web_test_user.id,
status=celery.states.STARTED,
queue_result_key=None,
show_in_ui=True
)
report_started2 = PersistentReport(
user_id=web_test_user.id,
status=celery.states.STARTED,
queue_result_key=None,
show_in_ui=True
)
report_finished = PersistentReport(
user_id=web_test_user.id,
status=celery.states.SUCCESS,
queue_result_key=None,
show_in_ui=True
)
self.session.add_all([
report_created,
report_started,
report_started2,
report_finished
])
self.session.commit()
#****************************************************************
# keep the test ids around so subclasses can use them
#****************************************************************
self.test_report_id = report_created.id
self.test_user_id = dan_user.id
self.test_web_user_id = web_test_user.id
self.test_cohort_id = test_cohort.id
self.test_cohort_name = test_cohort.name
self.test_cohort_user_id = dan_owns_test.id
self.test_wiki_user_id = dan.id
self.test_cohort_wiki_user_id = dan_in_test.id
self.test_logging_id = mw_logging.log_id
self.test_mediawiki_user_id = mw_user_dan.user_id
self.test_mediawiki_user_id_evan = mw_user_evan.user_id
self.test_mediawiki_user_id_andrew = mw_user_andrew.user_id
self.test_mediawiki_user_id_diederik = mw_user_diederik.user_id
self.test_page_id = mw_page.page_id
self.test_revision_id = rev1.rev_id
