mirror of
https://github.com/storytold/artcraft.git
synced 2026-11-05 17:58:55 +00:00
add a seed script that populates a lot more data
This commit is contained in:
1 parent
b6c54f23fa
commit
510c8ed09b
1 file changed
+70
Executable
+70
@@ -0,0 +1,70 @@
|
||||
#!/usr/bin/env python3
|
||||
|
||||
import requests
|
||||
import re
|
||||
import uuid
|
||||
|
||||
def print_response(response):
|
||||
print('\tStatus: {}'.format(response.status_code))
|
||||
for k, v in response.headers.items():
|
||||
print('\t header - {}: {}'.format(k, v))
|
||||
print('\tbody: {}', response.json())
|
||||
|
||||
# ==================== CREATE ACCOUNTS ====================
|
||||
|
||||
def create_user_account(username, password, email):
|
||||
print('Create account: {}'.format(username))
|
||||
signup_url = 'http://localhost:12345/create_account'
|
||||
payload = {
|
||||
'username': username,
|
||||
'password': password,
|
||||
'password_confirmation': password,
|
||||
'email_address': email,
|
||||
}
|
||||
r = requests.post(signup_url, json=payload)
|
||||
print_response(r)
|
||||
|
||||
create_user_account('brandon', 'testing', 'brandon@storyteller.company')
|
||||
create_user_account('echelon', 'testing', 'echelon@gmail.com')
|
||||
create_user_account('nanashi', 'nanashi', 'somebody@somewhere.com')
|
||||
|
||||
# ==================== LOGIN ====================
|
||||
|
||||
def login(username, password):
|
||||
print('Login: {}'.format(username))
|
||||
login_url = 'http://localhost:12345/login'
|
||||
payload = {
|
||||
'username_or_email': username,
|
||||
'password': password,
|
||||
}
|
||||
r = requests.post(login_url, json=payload)
|
||||
print_response(r)
|
||||
raw_set_cookie_header = r.headers['set-cookie']
|
||||
m = re.match(r'session=([^;\s]+);', raw_set_cookie_header)
|
||||
session_cookie = m.group(1)
|
||||
return session_cookie
|
||||
|
||||
session_echelon = login('echelon', 'testing')
|
||||
session_nanashi = login('nanashi', 'nanashi')
|
||||
|
||||
# ==================== UPLOAD DATA ====================
|
||||
|
||||
def upload_w2l_template(content_url, title, session_cookie):
|
||||
print('Upload W2L template')
|
||||
upload_url = 'http://localhost:12345/w2l/upload'
|
||||
cookies = { 'session': session_cookie }
|
||||
payload = {
|
||||
'idempotency_token': str(uuid.uuid4()),
|
||||
'title': title,
|
||||
'download_url': content_url,
|
||||
}
|
||||
r = requests.post(upload_url, cookies=cookies, json=payload)
|
||||
print_response(r)
|
||||
|
||||
|
||||
upload_w2l_template('https://i.imgur.com/W1EYiPU.jpg', 'foo', session_echelon)
|
||||
upload_w2l_template('https://i.imgur.com/NneE7Eq.jpg', 'bar', session_echelon)
|
||||
|
||||
upload_w2l_template('https://drive.google.com/file/d/1yoBSb6nwFUZVI4CD-nM2BSofGdBjC8Nn/view?usp=sharing', 'baz', session_nanashi)
|
||||
|
||||
|
||||
Reference in new issue
Block a user