def post(self): if self.request.get('_put'): self.put() return name = self.request.get('name') email = self.request.get('email') push_token = self.request.get('token') device_type = self.request.get('type') if not push_token or not device_type: api.write_error(self.response, 400, 'Missing required parameter, token or type') return user = User(uuid=str(uuid.uuid4()), auth=str(uuid.uuid4())) if name: user.name = name update_gender(user) if email: user.devices.append(Device(device_type='EMAIL', data=email)) user.devices.append(Device(device_type=device_type.upper(), data=push_token)) user.put() update_user_cards(user) api.search.update_public_index(user) api.write_message(self.response, 'success', extra={'users' : [get_user_json(user, public=False)]})
def __init__(self, user_id): if user_id is None: self.user_object = User.get_by_id(user_id) else: if not isinstance(user_id, str): raise ValueError("UserID should be string") try: self.user_object = User.get_by_id(user_id) if self.user_object is None: logging.warning("User not found, creating new user") self.user_object = User(id=user_id) self.user_object.put() except Exception as e: logging.error("Something wient wrong : {}".format(e)) raise Exception self.userid = self.user_object.key.id()
if weekly_avg_sleep and not math.isnan(weekly_avg_sleep): variables['weekly_average_sleep'] = str(weekly_avg_sleep) return variables def get_card_type(card): tags = list(card.tags) if 'archived' in tags: tags.remove('archived') tags.sort() return ','.join(tags) def get_card_week(card): week = None for tag in card.tags: if tag.startswith('week:'): try: return int(tag.split(':')[1]) except: pass return week if __name__ == '__main__': user = User() user.features.append(Pair(name='DUE_DATE_MILLIS', value=sys.argv[1])) for card in get_week_based_cards(user): print get_card_json(card)
MOCK_XSRF = MagicMock() MOCK_XSRF.XSRFProtect = NoOpDecorator sys.modules['xsrf'] = MOCK_XSRF import user FAKE_EMAIL = '*****@*****.**' FAKE_NAME = 'fake name' FAKE_PUBLIC_KEY = 'fakePublicKey' FAKE_PRIVATE_KEY = 'fakePrivateKey' FAKE_DS_KEY = 'urlEncodedKeyFromTheDatastore' FAKE_USER_KEY = ndb.Key(User, hashlib.sha256(FAKE_EMAIL).hexdigest()) FAKE_USER = User(key=FAKE_USER_KEY, email=FAKE_EMAIL, name=FAKE_NAME, public_key=FAKE_PUBLIC_KEY, private_key=FAKE_PRIVATE_KEY, is_key_revoked=False) FAKE_USER_ARRAY = [] FAKE_EMAIL_1 = u'*****@*****.**' FAKE_EMAIL_2 = u'*****@*****.**' FAKE_ADD_USER = {} FAKE_ADD_USER['primaryEmail'] = FAKE_EMAIL_1 FAKE_ADD_USER['name'] = {} FAKE_ADD_USER['name']['fullName'] = FAKE_NAME FAKE_ADD_USER['email'] = FAKE_EMAIL_1 FAKE_ADD_USER['role'] = 'MEMBER' FAKE_ADD_USER['type'] = 'USER' FAKE_USER_ARRAY.append(FAKE_ADD_USER)