From 82d2eee5070f76fd541dddc7710dcead00cdce43 Mon Sep 17 00:00:00 2001 From: vanzhiganov Date: Sat, 28 Nov 2015 01:33:32 +0300 Subject: [PATCH] cli app for add new data center --- cli-dc-add.py | 29 +++++++++++++++++++++++++++++ 1 file changed, 29 insertions(+) create mode 100644 cli-dc-add.py diff --git a/cli-dc-add.py b/cli-dc-add.py new file mode 100644 index 0000000..81a281a --- /dev/null +++ b/cli-dc-add.py @@ -0,0 +1,29 @@ +import argparse +from uuid import uuid4 +from hashlib import md5 +from app import models + +parser = argparse.ArgumentParser(description='') +parser.add_argument('--code', dest="code") +parser.add_argument('--name', dest="name") +parser.add_argument('--country', dest="country") +parser.add_argument('--city', dest="city") +parser.add_argument('--status', dest="status") + +args = parser.parse_args() + +dc_id = uuid4() + +if models.DataCenters.select().where( + models.DataCenters.code == args.code +).count() == 0: + models.DataCenters.create( + id=dc_id, + code=args.code, + name=args.name, + country=args.country, + city=args.city, + status=1 + ) +else: + print "Data center with code '%s' already exists." % args.code