xenofem/python-zulip-api
1
0
Fork 0
Code Issues Pull requests Packages Projects Releases Wiki Activity

mypy: Annotate tools/deploy using mypy

Browse source
This commit is contained in:
Shivam Gera 2018-03-17 17:38:04 +05:30 committed by showell
parent 8ef9b70191
commit 401674016c
1 changed files with 15 additions and 14 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

29
tools/deploy
View file

@ -7,15 +7,16 @@ import zipfile
import textwrap import textwrap
import requests import requests
import urllib import urllib
from urllib import parse
red = '\033[91m' red = '\033[91m' # type: str
green = '\033[92m' green = '\033[92m' # type: str
end_format = '\033[0m' end_format = '\033[0m' # type: str
bold = '\033[1m' bold = '\033[1m' # type: str
bots_dir = '.bots' bots_dir = '.bots' # type: str
def pack(options): def pack(options: argparse.Namespace) -> None:
# Basic sanity checks for input. # Basic sanity checks for input.
if not options.path: if not options.path:
print('tools/deploy: Path to bot folder not specified.') print('tools/deploy: Path to bot folder not specified.')
@ -59,7 +60,7 @@ def pack(options):
zip_file.close() zip_file.close()
print('pack: Created zip file at: {}.'.format(zip_file_path)) print('pack: Created zip file at: {}.'.format(zip_file_path))
def check_common_options(options): def check_common_options(options: argparse.Namespace) -> None:
if not options.server: if not options.server:
print('tools/deploy: URL to Botfarm server not specified.') print('tools/deploy: URL to Botfarm server not specified.')
sys.exit(1) sys.exit(1)
@ -67,7 +68,7 @@ def check_common_options(options):
print('tools/deploy: Botfarm deploy token not specified.') print('tools/deploy: Botfarm deploy token not specified.')
sys.exit(1) sys.exit(1)
def upload(options): def upload(options: argparse.Namespace) -> None:
check_common_options(options) check_common_options(options)
file_path = os.path.join(bots_dir, options.botname + '.zip') file_path = os.path.join(bots_dir, options.botname + '.zip')
if not os.path.exists(file_path): if not os.path.exists(file_path):
@ -86,7 +87,7 @@ def upload(options):
print('upload: Error {}. Aborting.'.format(r.status_code)) print('upload: Error {}. Aborting.'.format(r.status_code))
sys.exit(1) sys.exit(1)
def clean(options): def clean(options: argparse.Namespace) -> None:
file_path = os.path.join(bots_dir, options.botname + '.zip') file_path = os.path.join(bots_dir, options.botname + '.zip')
if os.path.exists(file_path): if os.path.exists(file_path):
os.remove(file_path) os.remove(file_path)
@ -94,7 +95,7 @@ def clean(options):
else: else:
print('clean: File \'{}\' not found.'.format(file_path)) print('clean: File \'{}\' not found.'.format(file_path))
def process(options): def process(options: argparse.Namespace) -> None:
check_common_options(options) check_common_options(options)
headers = {'key': options.key} headers = {'key': options.key}
url = urllib.parse.urljoin(options.server, 'bots/process') url = urllib.parse.urljoin(options.server, 'bots/process')
@ -109,7 +110,7 @@ def process(options):
print('process: Error {}: {}. Aborting.'.format(r.status_code, r.text)) print('process: Error {}: {}. Aborting.'.format(r.status_code, r.text))
sys.exit(1) sys.exit(1)
def start(options): def start(options: argparse.Namespace) -> None:
check_common_options(options) check_common_options(options)
headers = {'key': options.key} headers = {'key': options.key}
url = urllib.parse.urljoin(options.server, 'bots/start') url = urllib.parse.urljoin(options.server, 'bots/start')
@ -124,7 +125,7 @@ def start(options):
print('start: Error {}: {}. Aborting.'.format(r.status_code, r.text)) print('start: Error {}: {}. Aborting.'.format(r.status_code, r.text))
sys.exit(1) sys.exit(1)
def stop(options): def stop(options: argparse.Namespace) -> None:
check_common_options(options) check_common_options(options)
headers = {'key': options.key} headers = {'key': options.key}
url = urllib.parse.urljoin(options.server, 'bots/stop') url = urllib.parse.urljoin(options.server, 'bots/stop')
@ -139,13 +140,13 @@ def stop(options):
print('stop: Error {}: {}. Aborting.'.format(r.status_code, r.text)) print('stop: Error {}: {}. Aborting.'.format(r.status_code, r.text))
sys.exit(1) sys.exit(1)
def prepare(options): def prepare(options: argparse.Namespace) -> None:
pack(options) pack(options)
upload(options) upload(options)
clean(options) clean(options)
process(options) process(options)
def main(): def main() -> None:
usage = """tools/deploy <command> <bot-name> [options] usage = """tools/deploy <command> <bot-name> [options]
This is tool meant to easily deploy bots to a Zulip Bot Farm. This is tool meant to easily deploy bots to a Zulip Bot Farm.