Make executable Python code screenshots

python_photo.py

from bs4 import BeautifulSoup
from requests import Session
import sys
import zipfile


if len(sys.argv) >= 2:
    code = open(sys.argv[1]).read()
else:
    code = sys.stdin.read()

if not code:
    raise ValueError('Code must not be empty')


session = Session()

r = session.get('http://codephoto.ru/')
soup = BeautifulSoup(r.content, 'lxml')

csrf_token = soup.select_one('input[name="csrf_token"]')['value']

r = session.post('http://codephoto.ru/code', allow_redirects=False, data={
    'csrf_token': csrf_token,
    'language': 'Python',
    'code': code,
})

img_loc = r.headers['Location'].split('/')[-1]

r = session.get(f'http://codephoto.ru/upload/{img_loc}.jpg')

img = r.content


with zipfile.ZipFile('output.jpg', 'w') as out:
    out.writestr('__main__.py', code)

pyzip = open('output.jpg', 'rb').read()


with open('output.jpg', 'wb') as out:
    out.write(img)
    out.write(pyzip)