crowetic/protocol
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

contract_wrappers.py: fix ./setup.py clean

Browse Source
This commit is contained in:
F. Eugene Aumson
2019-08-08 16:20:24 -04:00
parent fc9c6c5434
commit 50f69f734f
1 changed files with 31 additions and 26 deletions
Download Patch File Download Diff File Expand all files Collapse all files

31
python-packages/contract_wrappers/setup.py
View File

@@ -18,15 +18,7 @@ from setuptools.command.test import test as TestCommand
BLACK_COMMAND = "black --line-length 79 "
class PreInstallCommand(distutils.command.build_py.build_py):
"""Custom setuptools command class for pulling in generated code."""
description = "Pull in code generated by TypeScript"
def run(self):
"""Copy files from TS build area to local src, & `black` them."""
pkgdir = path.dirname(path.realpath(argv[0]))
contracts = [
CONTRACTS_TO_BE_WRAPPED = [
"asset_proxy_owner",
"coordinator",
"coordinator_registry",
@@ -48,7 +40,17 @@ class PreInstallCommand(distutils.command.build_py.build_py):
"weth9",
"zrx_token",
]
for contract in contracts:
class PreInstallCommand(distutils.command.build_py.build_py):
"""Custom setuptools command class for pulling in generated code."""
description = "Pull in code generated by TypeScript"
def run(self):
"""Copy files from TS build area to local src, & `black` them."""
pkgdir = path.dirname(path.realpath(argv[0]))
for contract in CONTRACTS_TO_BE_WRAPPED:
copy(
path.join(
pkgdir,
@@ -84,7 +86,7 @@ class PreInstallCommand(distutils.command.build_py.build_py):
black_command = BLACK_COMMAND + " ".join(
[
f"src/zero_ex/contract_wrappers/{contract}/__init__.py"
for contract in contracts
for contract in CONTRACTS_TO_BE_WRAPPED
]
)
print(f"Running command `{black_command}`...")
@@ -158,8 +160,11 @@ class CleanCommandExtension(clean):
rmtree(".pytest_cache", ignore_errors=True)
rmtree("src/0x_contract_wrappers.egg-info", ignore_errors=True)
# generated files:
remove("src/zero_ex/contract_wrappers/exchange/__init__.py")
remove("src/zero_ex/contract_wrappers/erc20_token/__init__.py")
for contract in CONTRACTS_TO_BE_WRAPPED:
try:
remove(f"src/zero_ex/contract_wrappers/{contract}/__init__.py")
except FileNotFoundError:
pass
class TestPublishCommand(distutils.command.build_py.build_py):