amzlink.py

import os
import re

import pyperclip


def gen_affiliation_link(url):
    if not re.search(r"amazon.*/dp/", url):
        raise ValueError(f"{url} is not a valid Amazon product link")
    asin = re.sub(r".*/dp/([^/]+).*", r"\1", url)
    code = os.environ.get("AMAZON_AFFILIATE_CODE", "pyb0f-20")
    return f"http://www.amazon.com/dp/{asin}/?tag={code}"


def copy_affiliation_link():
    url = pyperclip.paste()
    link = gen_affiliation_link(url)
    pyperclip.copy(link)


if __name__ == '__main__':
    copy_affiliation_link()