def add_data(real_user, user, profile, **kw):
"""
can pass in data like 'phone:work'='555-555-5555'
"""
if not profile:
abort(403)
forbidden_keys = ['latitude', 'longitude']
class KVForm(formencode.Schema):
key = fv.UnicodeString(not_empty=True, min=2, max=64)
type = fv.OneOf([TYPE_WORK, TYPE_HOME, TYPE_MOBILE], not_empty=False)
value = fv.UnicodeString(not_empty=True, min=2, max=1024)
for k, v in kw.items():
type = None
s = k.split(':', 1)
if len(s) == 2:
k, type = s
if k in forbidden_keys: continue
if v:
scrubbed = validate(KVForm, key=k, value=v, type=type)
profile.add_data(user, scrubbed.key, scrubbed.value, type=scrubbed.type)
Session.flush()
return profile
def teather(real_user, user, profile=None, email=None, latitude=None, longitude=None, **kw):
"""
Will teather/follow another user
kw can be extra data to be asscoiated with the teather. Passing a type is done
by using a colon. i.e.
kw['phone'] = '415-343-1234' # would just add a phone number
kw['phone:work'] = '415-343-1234' # would add a phone number with type work
also, notes are associated the same way as everything else.
kw['notes'] = 'We met at the xyz blah'
"""
if (not profile and not email) or not user.profile:
abort(403)
profile = get(real_user, user, profile=profile, email=email)
if not profile and not email:
abort(403)
#create unclaimed
elif not profile:
email = validate(TeatherForm, email=email).email
profile = profiles.Profile(user=None)
Session.add(profile)
Session.flush()
add_data(real_user, user, profile, **kw)
#attach an email address...
if not email and profile.user:
email = profile.user.email
#all_data = profile.fetch_data()
#has_email = False
#for d in all_data:
# if d.key == 'email':
# has_email = True
# if (email and data.EmailHandler.normalize(email) != d.value):
# profile.add_data(user, 'email', email, type=d.type)
#if not has_email and not email:
# raise ClientException('Need to specify an email address!')
#elif not has_email:
if email:
profile.add_data(user, 'email', email)
t = user.profile.teather(profile, latitude=latitude, longitude=longitude)
Session.flush()
return t
def set_pref(real_user, user, key, value, use_real_user=True):
class Pref(formencode.Schema):
key = fv.MaxLength(64, not_empty=True)
value = fv.MaxLength(64, not_empty=False)
scrubbed = validate(Pref, key=key, value=value)
u = user
if use_real_user:
u = real_user
u.set_preference(scrubbed.key, scrubbed.value or '')
def create(user, **params):
"""
Creates a profile.
DO NOT EXPOSE THIS to the web api. Please.
"""
if user and user.profile:
raise AppException('User cannot have an existing profile.', field='user', code=INVALID)
params.setdefault('is_active', True)
scrubbed = validate(CreateForm, **params)
profile = profiles.Profile(user=user, **scrubbed)
Session.add(profile)
return profile
def create(**params):
"""
Creates a user.
DO NOT EXPOSE THIS to the web api. Please.
"""
numusers = len(Session.query(users.User).all())
scrubbed = validate(RegisterForm, **params)
user = users.User()
Session.add(user)
user.email = scrubbed.email
user.username = '******' in scrubbed and scrubbed.username or scrubbed.email
user.password = scrubbed.password
user.set_timezone_int(scrubbed.default_timezone)
if scrubbed.get('name'):
name = scrubbed.get('name').split(' ', 1)
user.first_name = name[0].strip()
user.last_name = len(name) == 2 and name[1].strip() or u''
else:
user.first_name = scrubbed.get('first_name')
user.last_name = scrubbed.get('last_name')
#first user is an admin.
if numusers == 0:
user.role = users.ROLE_ADMIN
#this will need some thought. We should have some kind of verification.
profile = profiles.Profile.find_unclaimed(user.email)
if profile:
profile.user = user
else:
from tethr import api
profile = api.profile.create(user)
if user.first_name or user.last_name:
profile.add_data(user, 'name', u'%s %s' % (user.first_name, user.last_name))
profile.add_data(user, 'email', user.email)
Session.flush()
return user
def explode_no_auth(real_user, user, type=None, **kw):
"""
An action to test the error handling of the stack. The tests use this.
"""
if type == 'app':
raise AppException('This is an app exception!', code=INVALID, field='type')
elif type == 'client':
raise ClientException('Oh Noes, ClientException!', code=INVALID, field='type')
elif type == 'client404':
raise ClientException('Oh Noes, ClientException NOT FOUND!', code=NOT_FOUND, field='type')
elif type == 'client403':
raise ClientException('Oh Noes, ClientException FORBIDDEN!', code=FORBIDDEN, field='type')
elif type == 'explosion':
1/0
elif type == 'http':
abort(404, 'This thing was not found!')
elif type == 'validation':
class Rawr(formencode.Schema):
meow = fv.Number()
scrubbed = validate(Rawr, meow='zzzz')
return kw
