Telethon/setup.py

228 lines
7.2 KiB
Python
Raw Normal View History

2017-06-21 20:18:22 +03:00
#!/usr/bin/env python3
"""A setuptools based setup module.
See:
https://packaging.python.org/en/latest/distributing.html
https://github.com/pypa/sampleproject
Extra supported commands are:
2018-04-15 14:19:25 +03:00
* gen, to generate the classes required for Telethon to run or docs
* pypi, to generate sdist, bdist_wheel, and push to PyPi
"""
import itertools
2018-06-06 18:35:06 +03:00
import json
2018-04-15 14:19:25 +03:00
import re
import shutil
2018-12-15 15:26:18 +03:00
from os import chdir
from pathlib import Path
from subprocess import run
2018-06-21 11:47:52 +03:00
from sys import argv
2016-11-30 00:29:42 +03:00
from setuptools import find_packages, setup
class TempWorkDir:
"""Switches the working directory to be the one on which this file lives,
while within the 'with' block.
"""
def __init__(self):
self.original = None
def __enter__(self):
2018-12-15 15:26:18 +03:00
self.original = Path('.')
chdir(str(Path(__file__).parent))
return self
def __exit__(self, *args):
chdir(str(self.original))
2018-12-15 15:26:18 +03:00
GENERATOR_DIR = Path('telethon_generator')
LIBRARY_DIR = Path('telethon')
2018-12-15 15:26:18 +03:00
ERRORS_IN = GENERATOR_DIR / 'data/errors.csv'
ERRORS_OUT = LIBRARY_DIR / 'errors/rpcerrorlist.py'
2018-12-15 15:26:18 +03:00
METHODS_IN = GENERATOR_DIR / 'data/methods.csv'
2018-06-06 18:35:06 +03:00
# Which raw API methods are covered by *friendly* methods in the client?
FRIENDLY_IN = GENERATOR_DIR / 'data/friendly.csv'
2018-12-15 15:26:18 +03:00
TLOBJECT_IN_TLS = [Path(x) for x in GENERATOR_DIR.glob('data/*.tl')]
TLOBJECT_OUT = LIBRARY_DIR / 'tl'
2018-04-15 14:19:25 +03:00
IMPORT_DEPTH = 2
2018-12-15 15:26:18 +03:00
DOCS_IN_RES = GENERATOR_DIR / 'data/html'
DOCS_OUT = Path('docs')
2018-04-15 14:19:25 +03:00
def generate(which, action='gen'):
from telethon_generator.parsers import\
parse_errors, parse_methods, parse_tl, find_layer
2018-04-15 14:19:25 +03:00
from telethon_generator.generators import\
generate_errors, generate_tlobjects, generate_docs, clean_tlobjects
2018-12-15 15:26:18 +03:00
layer = next(filter(None, map(find_layer, TLOBJECT_IN_TLS)))
errors = list(parse_errors(ERRORS_IN))
methods = list(parse_methods(METHODS_IN, FRIENDLY_IN, {e.str_code: e for e in errors}))
2018-12-15 15:26:18 +03:00
tlobjects = list(itertools.chain(*(
parse_tl(file, layer, methods) for file in TLOBJECT_IN_TLS)))
2018-04-15 14:19:25 +03:00
if not which:
which.extend(('tl', 'errors'))
2018-04-15 14:19:25 +03:00
2019-05-04 18:51:14 +03:00
clean = action == 'clean'
2018-04-15 14:19:25 +03:00
action = 'Cleaning' if clean else 'Generating'
if 'all' in which:
which.remove('all')
for x in ('tl', 'errors', 'docs'):
if x not in which:
which.append(x)
if 'tl' in which:
which.remove('tl')
print(action, 'TLObjects...')
if clean:
clean_tlobjects(TLOBJECT_OUT)
else:
generate_tlobjects(tlobjects, layer, IMPORT_DEPTH, TLOBJECT_OUT)
if 'errors' in which:
which.remove('errors')
print(action, 'RPCErrors...')
if clean:
2018-12-15 15:26:18 +03:00
if ERRORS_OUT.is_file():
ERRORS_OUT.unlink()
2018-04-15 14:19:25 +03:00
else:
with ERRORS_OUT.open('w') as file:
2018-04-15 14:19:25 +03:00
generate_errors(errors, file)
if 'docs' in which:
which.remove('docs')
print(action, 'documentation...')
if clean:
2018-12-15 15:26:18 +03:00
if DOCS_OUT.is_dir():
shutil.rmtree(str(DOCS_OUT))
2018-04-15 14:19:25 +03:00
else:
generate_docs(tlobjects, methods, layer, DOCS_IN_RES, DOCS_OUT)
2018-04-15 14:19:25 +03:00
2018-06-13 11:48:35 +03:00
if 'json' in which:
which.remove('json')
print(action, 'JSON schema...')
2018-12-15 15:26:18 +03:00
json_files = [x.with_suffix('.json') for x in TLOBJECT_IN_TLS]
2018-06-13 11:48:35 +03:00
if clean:
2018-12-15 15:26:18 +03:00
for file in json_files:
if file.is_file():
file.unlink()
2018-06-13 11:48:35 +03:00
else:
def gen_json(fin, fout):
2019-05-03 14:59:17 +03:00
meths = []
2018-06-13 11:48:35 +03:00
constructors = []
for tl in parse_tl(fin, layer):
if tl.is_function:
2019-05-03 14:59:17 +03:00
meths.append(tl.to_dict())
2018-06-13 11:48:35 +03:00
else:
constructors.append(tl.to_dict())
2019-05-03 14:59:17 +03:00
what = {'constructors': constructors, 'methods': meths}
2018-06-13 11:48:35 +03:00
with open(fout, 'w') as f:
json.dump(what, f, indent=2)
2019-05-03 14:59:17 +03:00
for fs in zip(TLOBJECT_IN_TLS, json_files):
gen_json(*fs)
2018-06-13 11:48:35 +03:00
2018-04-15 14:19:25 +03:00
if which:
print('The following items were not understood:', which)
print(' Consider using only "tl", "errors" and/or "docs".')
print(' Using only "clean" will clean them. "all" to act on all.')
print(' For instance "gen tl errors".')
def main():
2019-05-04 18:51:14 +03:00
if len(argv) >= 2 and argv[1] in ('gen', 'clean'):
generate(argv[2:], argv[1])
elif len(argv) >= 2 and argv[1] == 'pypi':
# (Re)generate the code to make sure we don't push without it
generate(['tl', 'errors'])
# Try importing the telethon module to assert it has no errors
try:
import telethon
except:
print('Packaging for PyPi aborted, importing the module failed.')
return
for x in ('build', 'dist', 'Telethon.egg-info'):
shutil.rmtree(x, ignore_errors=True)
run('python3 setup.py sdist', shell=True)
run('python3 setup.py bdist_wheel', shell=True)
run('twine upload dist/*', shell=True)
for x in ('build', 'dist', 'Telethon.egg-info'):
shutil.rmtree(x, ignore_errors=True)
else:
2018-04-15 14:19:25 +03:00
# e.g. install from GitHub
2018-12-15 15:26:18 +03:00
if GENERATOR_DIR.is_dir():
generate(['tl', 'errors'])
# Get the long description from the README file
with open('README.rst', 'r', encoding='utf-8') as f:
long_description = f.read()
with open('telethon/version.py', 'r', encoding='utf-8') as f:
version = re.search(r"^__version__\s*=\s*'(.*)'.*$",
f.read(), flags=re.MULTILINE).group(1)
setup(
2018-06-27 14:05:19 +03:00
name='Telethon',
version=version,
description="Full-featured Telegram client library for Python 3",
long_description=long_description,
url='https://github.com/LonamiWebs/Telethon',
download_url='https://github.com/LonamiWebs/Telethon/releases',
author='Lonami Exo',
author_email='totufals@hotmail.com',
license='MIT',
2018-04-04 11:21:55 +03:00
# See https://stackoverflow.com/a/40300957/4759433
# -> https://www.python.org/dev/peps/pep-0345/#requires-python
# -> http://setuptools.readthedocs.io/en/latest/setuptools.html
2018-06-21 11:47:52 +03:00
python_requires='>=3.5',
2018-04-04 11:21:55 +03:00
# See https://pypi.python.org/pypi?%3Aaction=list_classifiers
classifiers=[
# 3 - Alpha
# 4 - Beta
# 5 - Production/Stable
2018-06-27 14:05:19 +03:00
'Development Status :: 5 - Production/Stable',
'Intended Audience :: Developers',
'Topic :: Communications :: Chat',
'License :: OSI Approved :: MIT License',
'Programming Language :: Python :: 3',
'Programming Language :: Python :: 3.5',
'Programming Language :: Python :: 3.6'
],
keywords='telegram api chat client library messaging mtproto',
packages=find_packages(exclude=[
2018-06-21 11:47:52 +03:00
'telethon_*', 'run_tests.py', 'try_telethon.py'
]),
install_requires=['pyaes', 'rsa'],
extras_require={
2018-03-17 19:38:16 +03:00
'cryptg': ['cryptg']
}
)
if __name__ == '__main__':
with TempWorkDir(): # Could just use a try/finally but this is + reusable
main()