post_snippet.py

"""
Script to post a code snippet to PyBites CodeImag.es
If you see other use cases (e.g. process a csv file with snippets),
contact @bbelderbos on Twitter, thanks.
"""
from pprint import pprint as pp

# pip install requests python-decouple
import requests
from decouple import config

CODEIMAGES_USER = config("CODEIMAGES_USER")
CODEIMAGES_PASSWORD = config("CODEIMAGES_PASSWORD")
BASE_URL = "https://pybites-codeimages.herokuapp.com"
TOKEN_URL = f"{BASE_URL}/token"
CREATE_TIP_URL = f"{BASE_URL}/create"


def _write_multiline_input(action):
    print(f"{action}, enter <enter>+qq to finish: ")
    lines = []
    exit_mark = "qq"
    while True:
        line = input()
        if line.strip().lower() == exit_mark:
            break
        lines.append(line)
    return "\n".join(lines)


def get_token():
    payload = {"username": CODEIMAGES_USER, "password": CODEIMAGES_PASSWORD}
    resp = requests.post(TOKEN_URL, data=payload)
    token = resp.json()["access_token"]
    return token


def main():
    token = get_token()
    while True:
        title = input("Add a title: ")
        code = _write_multiline_input("Paste your code snippet")
        description = _write_multiline_input("Add an optional description")

        payload = {
            "title": title.strip(),
            "code": code.lstrip(),
            "description": description.strip(),
        }
        print("Posting tip ...")
        headers = {"Authorization": f"Bearer {token}"}
        resp = requests.post(CREATE_TIP_URL, json=payload, headers=headers)
        if str(resp.status_code).startswith("2"):
            print(f"Code snippet posted to CodeImag.es: {BASE_URL}")
        else:
            print(f"Something went wrong, API returned status code {resp.status_code}")
        print()
        print("API response:")
        pp(resp.json())

        if input("Press enter to post another tip, 'q' to exit: ") == "q":
            print("Bye")
            break


if __name__ == "__main__":
    main()