zulip-api-examples: Check and print path to script.

This commit is contained in:
derAnfaenger 2017-09-15 10:55:55 +02:00 committed by Tim Abbott
parent 6630deda6a
commit 091f532da6

View file

@ -2,14 +2,26 @@
import os import os
import zulip import zulip
import argparse
def main(): def main():
usage = """zulip-api-examples [script_name]
Prints the path to the Zulip API example scripts."""
parser = argparse.ArgumentParser(usage=usage)
parser.add_argument('script_name',
nargs='?',
default='',
help='print path to the script <script_name>')
args = parser.parse_args()
zulip_path = os.path.abspath(os.path.dirname(zulip.__file__)) zulip_path = os.path.abspath(os.path.dirname(zulip.__file__))
examples_path = os.path.abspath(os.path.join(zulip_path, 'examples')) examples_path = os.path.abspath(os.path.join(zulip_path, 'examples', args.script_name))
if os.path.isdir(examples_path): if os.path.isdir(examples_path) or (args.script_name and os.path.isfile(examples_path)):
print(examples_path) print(examples_path)
else: else:
raise OSError("Examples cannot be accessed at {}: Directory does not exist!" raise OSError("Examples cannot be accessed at {}: {} does not exist!"
.format(examples_path)) .format(examples_path,
"File" if args.script_name else "Directory"))
if __name__ == '__main__': if __name__ == '__main__':
main() main()