def incrementStat(statname,value): stats = Stat.objects(name=statname) stat = None if not stats or len(stats)<=0: stat = Stat() stat.name = statname stat.total = 0 else: stat = stats[0] stat.total = stat.total + value stat.save()
def create_new_user(user_info): new_user = User(username=user_info['username'], bio=user_info['bio'], website=user_info['website'], profile_picture=user_info['profile_picture'], user_id=user_info['id'], full_name=user_info['full_name']) user_stat = Stat(followed_by=user_info['counts']['followed_by'], follows=user_info['counts']['follows'], media=user_info['counts']['media']) new_user.stat = user_stat user_stat.save() new_user.save() return new_user
def test_parse_stat(self): response = ''' {"response":[[ {"month":"2011-02","impressions":0,"clicks":0,"money":"0.00"}, {"month":"2011-03","impressions":123456,"clicks":789,"money":"123.45"} ]]} ''' account = Account.objects.create(remote_id=1) instance = Stat(account=account, period=0, data={}) instance.parse(json.loads(response)['response'][0][1]) instance.save() self.assertEqual(instance.clicks, 789) self.assertEqual(instance.impressions, 123456) self.assertEqual(instance.month, '2011-03') self.assertEqual(instance.money, '123.45')