server.py

from flask import Flask
from flask import request
from flask import jsonify

import boto3

from yowsup.common.tools import StorageTools, WATools
from yowsup.registration.existsrequest import WAExistsRequest
from yowsup.env import S40YowsupEnv, AndroidYowsupEnv

import random, hashlib, os
from urllib import quote
from collections import defaultdict

CURRENT_ENV = AndroidYowsupEnv()

app = Flask(__name__)

@app.route("/", methods=['GET'])
def index():
  return jsonify(status=200)

@app.route("/token", methods=['GET'])
def token():
  number = request.args.get('number')
  cc = request.args.get('cc')


  if number is not None:
    if cc is not None:

      in_ = number[len(cc):]
      token = CURRENT_ENV.getToken(in_)
      idx = WATools.generateIdentity()

      return jsonify(status=200, number=number, cc=cc, token=token, identity=quote(idx))
    else:
      return jsonify(status=402, error="Country code is required")
  else:
    return jsonify(status=402, error="Number is required")

@app.route("/instances", methods=['GET'])
def instances():
  ec2 = boto3.resource('ec2')

  purpose_filter = { 'Name': 'tag:Purpose', 'Values': ['container'] }
  running_instances = ec2.instances.filter(Filters=[ purpose_filter ])

  details = defaultdict()

  servers = {
    'count': 0,
    'instances': []
  }

  for instance in running_instances:
    details = {
      'instance_id': instance.id,
      'state': instance.state['Name'],
      'public_ip': instance.public_ip_address
    }

    servers['instances'].append(details)
    servers['count'] += 1

  return jsonify(status=200, servers=servers)


# Get an environment variable
def get_env(key, raiseError=True, default_value=None):
  value = os.environ.get(key)
  if value is None:
    if raiseError:
      raise Exception("Error. Environment Variables not loaded, kindly load them " % key)
    else:
      return default_value
  else:
    return value.encode('utf-8')

if __name__ == "__main__":
  debug = get_env('debug', False, False)
  app.run(debug=debug)