mirror of
https://github.com/trezor/trezor-firmware.git
synced 2024-11-30 03:18:20 +00:00
137 lines
3.9 KiB
Python
Executable File
137 lines
3.9 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
import glob
|
|
import os.path
|
|
import re
|
|
import shutil
|
|
import subprocess
|
|
import sys
|
|
import tempfile
|
|
|
|
from setuptools import setup, Command, find_packages
|
|
from setuptools.command.build_py import build_py
|
|
from setuptools.command.develop import develop
|
|
from distutils.errors import DistutilsError
|
|
|
|
install_requires = [
|
|
'setuptools>=19.0',
|
|
'ecdsa>=0.9',
|
|
'mnemonic>=0.17',
|
|
'requests>=2.4.0',
|
|
'click>=6.2',
|
|
'pyblake2>=0.9.3',
|
|
'libusb1>=1.6.4',
|
|
'construct>=2.9',
|
|
]
|
|
|
|
CWD = os.path.dirname(os.path.realpath(__file__))
|
|
TREZOR_COMMON = os.path.join(CWD, 'vendor', 'trezor-common')
|
|
|
|
|
|
def read(*path):
|
|
filename = os.path.join(CWD, *path)
|
|
with open(filename, 'r') as f:
|
|
return f.read()
|
|
|
|
|
|
def find_version():
|
|
version_file = read("trezorlib", "__init__.py")
|
|
version_match = re.search(r"^__version__ = \"(.*)\"$", version_file, re.M)
|
|
if version_match:
|
|
return version_match.group(1)
|
|
else:
|
|
raise RuntimeError("Version string not found")
|
|
|
|
|
|
class PrebuildCommand(Command):
|
|
description = 'update vendored files (coins.json, protobuf messages)'
|
|
user_options = []
|
|
|
|
def initialize_options(self):
|
|
pass
|
|
|
|
def finalize_options(self):
|
|
pass
|
|
|
|
def run(self):
|
|
# check for existence of the submodule directory
|
|
common_defs = os.path.join(TREZOR_COMMON, 'defs')
|
|
if not os.path.exists(common_defs):
|
|
raise DistutilsError('trezor-common submodule seems to be missing.\n' +
|
|
'Use "git submodule update --init" to retrieve it.')
|
|
|
|
# generate and copy coins.json to the tree
|
|
with tempfile.TemporaryDirectory() as tmpdir:
|
|
build_coins = os.path.join(TREZOR_COMMON, 'tools', 'build_coins.py')
|
|
subprocess.check_call([sys.executable, build_coins], cwd=tmpdir)
|
|
shutil.copy(os.path.join(tmpdir, 'coins.json'), os.path.join(CWD, 'trezorlib', 'coins.json'))
|
|
|
|
# regenerate messages
|
|
try:
|
|
proto_srcs = glob.glob(os.path.join(TREZOR_COMMON, "protob", "*.proto"))
|
|
subprocess.check_call([
|
|
sys.executable,
|
|
os.path.join(TREZOR_COMMON, "protob", "pb2py"),
|
|
"-o", os.path.join(CWD, "trezorlib", "messages"),
|
|
"-P", "..protobuf",
|
|
] + proto_srcs)
|
|
except Exception as e:
|
|
raise DistutilsError("Generating protobuf failed. Make sure you have 'protoc' in your PATH.") from e
|
|
|
|
|
|
def _patch_prebuild(cls):
|
|
"""Patch a setuptools command to depend on `prebuild`"""
|
|
orig_run = cls.run
|
|
|
|
def new_run(self):
|
|
self.run_command('prebuild')
|
|
orig_run(self)
|
|
|
|
cls.run = new_run
|
|
|
|
|
|
_patch_prebuild(build_py)
|
|
_patch_prebuild(develop)
|
|
|
|
|
|
setup(
|
|
name='trezor',
|
|
version=find_version(),
|
|
author='TREZOR',
|
|
author_email='info@trezor.io',
|
|
license='LGPLv3',
|
|
description='Python library for communicating with TREZOR Hardware Wallet',
|
|
long_description='{}\n\n{}'.format(
|
|
read('README.md'),
|
|
read('CHANGELOG.md'),
|
|
),
|
|
long_description_content_type='text/markdown',
|
|
url='https://github.com/trezor/python-trezor',
|
|
packages=find_packages(),
|
|
package_data={
|
|
'trezorlib': ['coins.json'],
|
|
},
|
|
scripts=['trezorctl'],
|
|
install_requires=install_requires,
|
|
extras_require={
|
|
':python_version < "3.5"': ['typing>=3.0.0'],
|
|
'hidapi': ['hidapi>=0.7.99.post20'],
|
|
'ethereum': [
|
|
'rlp>=0.4.4',
|
|
'ethjsonrpc>=0.3.0',
|
|
],
|
|
},
|
|
python_requires='>=3.3',
|
|
include_package_data=True,
|
|
zip_safe=False,
|
|
classifiers=[
|
|
'License :: OSI Approved :: GNU Lesser General Public License v3 (LGPLv3)',
|
|
'Operating System :: POSIX :: Linux',
|
|
'Operating System :: Microsoft :: Windows',
|
|
'Operating System :: MacOS :: MacOS X',
|
|
'Programming Language :: Python :: 3 :: Only',
|
|
],
|
|
cmdclass={
|
|
'prebuild': PrebuildCommand,
|
|
},
|
|
)
|