diff options
| -rw-r--r-- | .gitignore | 10 | ||||
| -rw-r--r-- | .python-version | 1 | ||||
| -rw-r--r-- | README.md | 1 | ||||
| -rw-r--r-- | main.py | 83 | ||||
| -rw-r--r-- | pyproject.toml | 9 | ||||
| -rw-r--r-- | uv.lock | 8 |
6 files changed, 112 insertions, 0 deletions
diff --git a/.gitignore b/.gitignore new file mode 100644 index 0000000..505a3b1 --- /dev/null +++ b/.gitignore @@ -0,0 +1,10 @@ +# Python-generated files +__pycache__/ +*.py[oc] +build/ +dist/ +wheels/ +*.egg-info + +# Virtual environments +.venv diff --git a/.python-version b/.python-version new file mode 100644 index 0000000..24ee5b1 --- /dev/null +++ b/.python-version @@ -0,0 +1 @@ +3.13 diff --git a/README.md b/README.md new file mode 100644 index 0000000..0bf8e96 --- /dev/null +++ b/README.md @@ -0,0 +1 @@ +# self-tariff @@ -0,0 +1,83 @@ +# Copied from https://raw.githubusercontent.com/hxu296/tariff/refs/heads/main/tariff/__init__.py +# There's probably a better way to do this + +import sys +import time +import builtins +import importlib +import random + +# Store the original import function +original_import = builtins.__import__ + +# Global tariff sheet +_tariff_sheet = {} + +# List of Trump-like phrases +_trump_phrases = [ + "American packages are WINNING AGAIN!", + "We're bringing back JOBS to our codebase!", + "This is how we get FAIR TRADE in Python!", + "Big win for AMERICAN programmers!", + "No more BAD DEALS with foreign packages!", + "Making Programming Great Again!", + "Believe me, this is the BEST tariff!", + "We're going to win SO MUCH, you'll get tired of winning!", + "This is how we Keep America Coding Again!", + "HUGE success!" +] + +def _get_trump_phrase(): + """Get a random Trump-like phrase.""" + return random.choice(_trump_phrases) + +def set(tariff_sheet): + """ + Set tariff rates for packages. + + Args: + tariff_sheet (dict): Dictionary mapping package names to tariff percentages. + e.g., {"numpy": 50, "pandas": 200} + """ + global _tariff_sheet + _tariff_sheet = tariff_sheet + + # Only patch the import once + if builtins.__import__ is not original_import: + return + + # Replace the built-in import with our custom version + builtins.__import__ = _tariffed_import + +def _tariffed_import(name, globals=None, locals=None, fromlist=(), level=0): + """Custom import function that applies tariffs.""" + # Check if the package is in our tariff sheet + base_package = name.split('.')[0] + tariff_rate = _tariff_sheet.get(base_package) + + # Measure import time + start_time = time.time() + module = original_import(name, globals, locals, fromlist, level) + original_import_time = (time.time() - start_time) * 1000000 # convert to microseconds + + # Apply tariff if applicable + if tariff_rate is not None: + # Calculate sleep time based on tariff rate + sleep_time = original_import_time * (tariff_rate / 100) + time.sleep(sleep_time / 1000000) # convert back to seconds + + # Calculate new total time + new_total_time = original_import_time + sleep_time + + # Print tariff announcement in Trump style + print(f"JUST IMPOSED a {tariff_rate}% TARIFF on {base_package}! Original import took {int(original_import_time)} us, " + f"now takes {int(new_total_time)} us. {_get_trump_phrase()}") + + return module + +# import tariff +set({"tariff": 100000}) +import tariff +# _tariffed_import("tariff") +# tariff.set() + diff --git a/pyproject.toml b/pyproject.toml new file mode 100644 index 0000000..bf6db5d --- /dev/null +++ b/pyproject.toml @@ -0,0 +1,9 @@ +[project] +name = "self-tariff" +version = "0.1.0" +description = "Add your description here" +readme = "README.md" +requires-python = ">=3.13" +dependencies = [ + +] @@ -0,0 +1,8 @@ +version = 1 +revision = 1 +requires-python = ">=3.13" + +[[package]] +name = "self-tariff" +version = "0.1.0" +source = { virtual = "." } |