proxy_api.py

import json
from urllib import request

import pandas as pd
import numpy as np
from flask import Flask
from flask import jsonify, render_template

app = Flask(__name__,template_folder='./static')
FILTER = True
# piracicabana
# ORIGINAL_URL='http://00224.transdatasmart.com.br:22401/ITS-infoexport/api/Data/VeiculosGTFS'

# pioneira
ORIGINAL_URL='https://www.sistemas.dftrans.df.gov.br/service/gps/operacoes'

def get_data():
    req = request.Request(ORIGINAL_URL, method='GET')
    app.logger.info('Requesting on %s' % ORIGINAL_URL)

    response = request.urlopen(req)
    body = response.read().decode('utf-8')
    body = json.loads(body)

#    df = pd.DataFrame.from_records(body['Dados'], columns=body['Campos'])
    df = pd.DataFrame()
    for operadora in body:
        df_temp = pd.DataFrame.from_records(operadora['veiculos'])
        df = pd.concat([df,df_temp], ignore_index=True)
    
    df = process_data(df)

    if(FILTER):
        df = apply_filters(df)

    return df.to_dict(orient='records')

def process_data(df):
    #df = df.apply(convert_lat_long, axis=1)
    df_lat_long = df.localizacao.apply(pd.Series)
    df = df.merge(df_lat_long, on=df.index)
    
    # limpa campos com string vazia -> ""
    df['GPS_Latitude'] = df['latitude']
    df['GPS_Longitude'] = df['longitude']

    df['GPS_Latitude'].replace('', np.nan, inplace=True)
    df['GPS_Longitude'].replace('', np.nan, inplace=True)
    df = df.dropna(subset=['GPS_Latitude', 'GPS_Longitude'])

    df.GPS_Latitude = df.GPS_Latitude.astype(float)
    df.GPS_Longitude = df.GPS_Longitude.astype(float)
    df.drop(columns=['latitude','longitude','localizacao'], inplace=True)

    return df

def apply_filters(df):
    
    linhas = ['0.195', '147.5', '147.6', '180.1', '180.2', '181.2', '181.4', '8002','106.2','0.147','2207','2209']
    
    df = df[df.linha.isin(linhas)]
    
    return df

@app.route('/')
def root():
    #return json.dumps({'success':True}), 200, {'ContentType':'application/json'} 
    return render_template('index.html')
    


@app.route("/data")
def proxy():

    original_resp = get_data()
    response = jsonify(original_resp)

    return response, 200, {'ContentType':'application/json'}

@app.route('/teste')
def teste():
    return json.dumps({'success':True}), 200, {'ContentType':'application/json'} 

import os
port = int(os.environ.get("PORT", 5000))
app.run(host='0.0.0.0', port=port)