diff --git a/debug/curl/seed.py b/debug/curl/seed.py new file mode 100755 index 0000000000..3e65c2c017 --- /dev/null +++ b/debug/curl/seed.py @@ -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) + +