python-zulip-api/zulip_bots/generate_manifest.py
Eeshan Garg dfe4360108 zulip_bots/MANIFEST: Support generating MANIFEST for PyPA release.
The generate_manifest.py script now takes a --release argument
that generates a separate MANIFEST for a PyPA release that
include:

* doc.md files
* related assets/* files

A release MANIFEST doesn't include logos and fixtures.
A non-release MANIFEST includes everything.
2017-10-05 12:01:37 -07:00

99 lines
2.7 KiB
Python
Executable file

#!/usr/bin/env python
import argparse
import os
import glob
CURRENT_DIR = os.path.abspath(os.path.dirname(__file__))
BOTS_DIR = os.path.normpath(os.path.join(CURRENT_DIR, 'zulip_bots', 'bots'))
MANIFEST_PATH = os.path.join(CURRENT_DIR, 'MANIFEST.in')
def get_test_fixtures():
# type: () -> List[str]
glob_pattern = os.path.join(BOTS_DIR, '*', 'fixtures', '*.json')
fixtures_paths = map(
lambda fp: os.path.join(*fp.split(os.path.sep)[-5:]).replace(os.path.sep, '/'),
glob.glob(glob_pattern)
)
return fixtures_paths
def get_logos():
# type: () -> List[str]
glob_pattern = os.path.join(BOTS_DIR, '*', 'logo.*')
logo_paths = map(
lambda fp: os.path.join(*fp.split(os.path.sep)[-4:]).replace(os.path.sep, '/'),
glob.glob(glob_pattern)
)
return logo_paths
def get_docs():
# type: () -> List[str]
glob_pattern = os.path.join(BOTS_DIR, '*', 'doc.md')
doc_paths = map(
lambda fp: os.path.join(*fp.split(os.path.sep)[-4:]).replace(os.path.sep, '/'),
glob.glob(glob_pattern)
)
return doc_paths
def get_assets():
# type: () -> List[str]
glob_pattern = os.path.join(BOTS_DIR, '*', 'assets', '*')
assets_files = map(
lambda fp: os.path.join(*fp.split(os.path.sep)[-5:]).replace(os.path.sep, '/'),
glob.glob(glob_pattern)
)
return assets_files
def generate_and_write(filepaths, file_obj):
# type: () -> None
template = 'include {line}\n'
lines = map(lambda line: template.format(line=line), filepaths)
file_obj.writelines(lines)
file_obj.write('\n')
def generate_dev_manifest():
# type: () -> None
with open(MANIFEST_PATH, 'w') as fp:
generate_and_write(get_test_fixtures(), fp)
generate_and_write(get_logos(), fp)
generate_and_write(get_docs(), fp)
generate_and_write(get_assets(), fp)
def generate_release_manifest():
# type: () -> None
with open(MANIFEST_PATH, 'w') as fp:
generate_and_write(get_docs(), fp)
generate_and_write(get_assets(), fp)
def parse_args():
usage = """
To generate a MANIFEST.in for a PyPA release, run:
./generate_manifest.py --release
To generate a MANIFEST.in for development, run without arguments:
./generate_manifest.py
"""
parser = argparse.ArgumentParser(usage=usage)
parser.add_argument('--release', '-r',
action='store_true',
default=False,
help='Generate MANIFEST.in for a PyPA release.')
return parser.parse_args()
def main():
# type: () -> None
options = parse_args()
if options.release:
generate_release_manifest()
else:
generate_dev_manifest()
if __name__ == '__main__':
main()