generator.py

import json, requests, urllib2, random
from werkzeug.security import generate_password_hash
from loremipsum import generate_paragraph
import datetime
from bson import json_util
import json
import time

print '---------------------------------fetching users-----------------------'
url = 'http://127.0.0.1:8000/api/people'
randomuser = urllib2.urlopen('http://api.randomuser.me/?results=10&seed='+str(random.randrange(0, 101, 2)))
results = json.loads(randomuser.read())
users=results['results']
processed_users = []
headers = {'content-type': 'application/json'}
dt = str(datetime.datetime.now())
ts = int(time.time())

for raw_user in users:
    user = raw_user['user']
    user.pop('sha1')
    user.pop('sha256')
    user.pop('registered')
    user.pop('md5')
    user.pop('salt')
    user.pop('SSN')
    user.pop('cell')
    user.pop('version')
    user.pop('nationality')
    user['born'] = user.pop('dob')
    user['email_confirmed'] = True
    user['lastmessageseen'] = ts
    user['interests'] = []
    user['study'] = {
        'intermediate':"",
        'graduate':""
    }
    user['password'] = {
        'password':generate_password_hash('111'),
        'password_test': '111',
        'password_updated':dt
    }
    user['movies'] = []
    user['send_add_requests'] = []
    user['token'] = ''
    user['friends'] = []
    user['notifications'] = []
    user['conversations'] = []
    user['role'] = "test"
    user['questions']=[]
    user['study'] = {"school": "my school",
                     "graduate":"Engineering"
                     }

    print '----------------------------user-------------------------'
    print user
    r = requests.post(url, data=json.dumps(user, default=json_util.default), headers=headers)
    print '====================================================='
    print r.content
    resp = json.loads(r.content)
    processed_users.append({'id': resp['_id'], 'etag': resp['_etag']})
print '---------------------------------Adding friends to users-----------------------'
print processed_users
uids = [x['id'] for x in processed_users]
for user in processed_users:
		friends=set(random.sample(uids,int(random.random()*10)))
		friends-set([user['id']])
		headers['If-Match'] = user['etag']
		friends_dict = {}
		#friends_dict['friends'] = []
		print friends_dict
		print headers
		#r = requests.patch(url + '/' + user['id'], data=json.dumps(friends_dict), headers=headers)
		print r.content
print '---------------------------------Adding posts to users-----------------------'
for user in processed_users:
    for _ in range(20):
        post={}
        post['type'] = 'text'
        post['author'] = user['id']
        post['content'] = generate_paragraph()[2]
        post['interestedPeople'] = []
        r = requests.post(url + '/' + user['id'] + '/posts', data=json.dumps(post), headers={'content-type': 'application/json'})
        print r.content