def populate_db():
for i in range(15):
target = UserProfile.create(
username=f'test{i}',
password=passw,
name=f'test#{i}',
email=f'fort{i}@gma.com',
confirmation_sent_at=datetime.datetime.now(),
last_sign_in_at=1)
print(f"Created user {target.username}")
target.follow(yab)
print("Following user")
#crate some followers
for j in range(i):
if j != i:
user = UserProfile.get(username=f'test{j}')
print(f'{user.username} -> {target.username}')
user.follow(target, True)
#craete some images via API
print("Uploading some pics")
for j in range(PHOTOS):
print(f'Image {j} of {PHOTOS}')
"""
def populate_for_travis():
for i in range(15):
target = UserProfile.create(
username=f'test{i}',
password=passw,
name=f'test#{i}',
email=f'fort{i}@gma.com',
confirmation_sent_at=datetime.datetime.now(),
last_sign_in_at=1)
target.follow(yab)
#crate some followers
for j in range(i):
if j != i:
user = UserProfile.get(username=f'test{j}')
user.follow(target, True)
def populate_db():
for i in range(15):
target = UserProfile.create(
username=f'test{i}',
password=passw,
name=f'test#{i}',
email=f'fort{i}@gma.com',
confirmation_sent_at=datetime.datetime.now(),
last_sign_in_at=1)
print(f"Created user {target.username}")
target.follow(yab)
print("Following user")
#crate some followers
for j in range(i):
if j != i:
user = UserProfile.get(username=f'test{j}')
print(f'{user.username} -> {target.username}')
user.follow(target, True)
#craete some images via API
print("Uploading some pics")
for j in range(PHOTOS):
print(f'Image {j} of {PHOTOS}')
client = testing.TestClient(app)
data = {'username': target.username, 'password': '******'}
header = {
'Authorization':
"Basic " + b64encode(f"{data['username']}:{data['password']}".
encode()).decode('utf-8')
}
token = client.simulate_get("/api/v1/auth",
json=data,
headers=header)
token = token.json['token']
auth = {'Authorization': token}
images = ['fine.jpg', 'wildunix.jpeg', 'gatos.jpeg']
image = random.choice(images)
ids = []
data, headers = create_multipart(
open(f'tests/assets/{image}', 'rb').read(),
fieldname="file",
filename='image',
content_type=f'image/{image.split(".")[1]}')
auth.update(headers)
response = client.simulate_post("/api/v1/media",
body=data,
headers=auth)
ids.append(response.json['id'])
status = {
'visibility': True,
'status': f'My photo #{j} by @{target.username}',
'sensitive': random.choice([True, False]),
'media_ids': ','.join(ids)
}
response = client.simulate_post("/api/v1/statuses",
params=status,
headers=auth)
from manage_db import connect
from models.followers import FollowerRelation
from models.user import UserProfile
from models.status import Status
connect()
a = FollowerRelation.select()
print("Follows:")
for u in a:
print(u)
print("====================")
print("Users:")
for u in UserProfile.select():
print(u.username, " con ap_id ", str(u.ap_id))
print(list(u.timeline()))
print("----")
print("====================")
print(UserProfile.select().count())
print("========")
print("Statuss")
for p in Status.select():
print(p)
print("========")
print("Following test:")
t = UserProfile.get(username="******")
print(t.followers().count())