Removed unecessary use of %.
#!/usr/bin/python2.5
#
# Copyright 2009 the Melange authors.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
"""Cron job handler for adding unique user id.
"""
__authors__ = [
'"Pawel Solyga" <pawel.solyga@gmail.com>',
]
from google.appengine.ext import db
from google.appengine.api import users
from soc.logic.models.job import logic as job_logic
from soc.logic.models.priority_group import logic as priority_logic
from soc.logic.models.user import logic as user_logic
# amount of users to create jobs for before updating
DEF_USER_STEP_SIZE = 10
class TempUserWithUniqueId(db.Model):
"""Helper model for temporary storing User Property with unique id.
"""
user = db.UserProperty(required=True)
def emailToAccountAndUserId(address):
"""Return a stable user_id string based on an email address, or None if
the address is not a valid/existing google account.
"""
user = users.User(address)
key = TempUserWithUniqueId(user=user).put()
obj = TempUserWithUniqueId.get(key)
# pylint: disable-msg=E1103
return (obj, obj.user.user_id())
def setupUniqueUserIdAdder(job_entity):
"""Job that setup jobs that will add unique user ids to all Users.
Args:
job_entity: a Job entity with key_data set to
[last_completed_user]
"""
key_data = job_entity.key_data
user_fields = {'user_id': None}
if len(key_data) == 1:
# start where we left off
user_fields['__key__ >'] = key_data[0]
m_users = user_logic.getForFields(user_fields,
limit=DEF_USER_STEP_SIZE)
# set the default fields for the jobs we are going to create
priority_group = priority_logic.getGroup(priority_logic.CONVERT)
job_fields = {
'priority_group': priority_group,
'task_name': 'addUniqueUserIds'}
job_query_fields = job_fields.copy()
while m_users:
# for each user create a adder job
for user in m_users:
job_query_fields['key_data'] = user.key()
adder_job = job_logic.getForFields(job_query_fields, unique=True)
if not adder_job:
# this user doesn't have unique id yet
job_fields['key_data'] = [user.key()]
job_logic.updateOrCreateFromFields(job_fields)
# update our own job
last_user_key = m_users[-1].key()
if len(key_data) == 1:
key_data[0] = last_user_key
else:
key_data.append(last_user_key)
updated_job_fields = {'key_data': key_data}
job_logic.updateEntityProperties(job_entity, updated_job_fields)
# rinse and repeat
user_fields['__key__ >'] = last_user_key
m_users = user_logic.getForFields(user_fields,
limit=DEF_USER_STEP_SIZE)
# we are finished
return
def addUniqueUserIds(job_entity):
"""Job that will add unique user id to a User.
Args:
job_entity: a Job entity with key_data set to [user_key]
"""
from soc.cron.job import FatalJobError
user_keyname = job_entity.key_data[0].name()
user_entity = user_logic.getFromKeyName(user_keyname)
if not user_entity:
raise FatalJobError('The User with keyname %s does not exist!' % (
user_keyname))
# add unique user id
account, user_id = emailToAccountAndUserId(user_entity.account.email())
user_entity.account = account
user_entity.user_id = user_id
user_entity.put()
# we are done here
return