subnet and imsi endpoints
Change-Id: Ief1d09026c99559d12eb5c905e484ed9eb5f5e7e
diff --git a/networkdiag.py b/networkdiag.py
index 7529eae..cb8bedb 100644
--- a/networkdiag.py
+++ b/networkdiag.py
@@ -7,25 +7,58 @@
from flask import Flask, request
from flask_restful import Resource, Api
import logging as log
+from argparse import ArgumentParser, SUPPRESS
+
+from roc import Roc
app = Flask(__name__)
api = Api(app)
+global roc
-class Greeting(Resource):
+
+class Imsi(Resource):
def get(self):
- return 'Hello World!'
+ return roc.get_imsi()
+
+class Subnet(Resource):
+ def get(self):
+ return roc.get_subnet()
-api.add_resource(Greeting, '/') # Route_1
+api.add_resource(Imsi, '/imsi')
+api.add_resource(Subnet, '/subnet')
+
+
+def build_argparser():
+ parser = ArgumentParser(add_help=False)
+ args = parser.add_argument_group('Options')
+ args.add_argument('-h', '--help',
+ action='help',
+ default=SUPPRESS,
+ help='Show this help message and exit.')
+ args.add_argument("--user",
+ help="ROC username",
+ type=str)
+ args.add_argument("--password",
+ help="ROC password",
+ type=str)
+ return parser
+
if __name__ == '__main__':
log.basicConfig(
- format='%(asctime)s %(levelname)-8s %(message)s',
- level=log.DEBUG,
- datefmt='%Y-%m-%d %H:%M:%S',
- stream=sys.stdout)
+ format='%(asctime)s %(levelname)-8s %(message)s',
+ level=log.DEBUG,
+ datefmt='%Y-%m-%d %H:%M:%S',
+ stream=sys.stdout)
log.info("Starting network-diag-app...")
+ args = build_argparser().parse_args()
+
+ roc = Roc(args.user, args.password)
+ #roc.dump()
+ print(roc.get_imsi())
+
app.run('0.0.0.0', '3333')