zulip-api-examples: Check and print path to script.
This commit is contained in:
parent
6630deda6a
commit
091f532da6
|
@ -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()
|
||||||
|
|
Loading…
Reference in a new issue