project_m1/main.py at main · SilviaAlconGarrido/project_m1 · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60

import argparse
import pandas as pd
from fuzzywuzzy import process
#Aqui elegimos entre bicimad y bicipark. Dependiendo de la elección imprime el csv entero
def resumen(categoria):
    df_biciMadf =  pd.read_csv('./data/bici_monu.csv')
    df_biciParkf = pd.read_csv('./data/bicipark_monu.csv')
    if categoria == 'Bicimad':
        print(f'Elegiste: {categoria}')
        print(df_biciMadf)
    elif categoria == 'Bicipark':
        print(f'Elegiste: {categoria}')
        print(df_biciMadf)


#Esto me hace elegir entre Bicimad y bicipark pediendome el sitio
def elegir(categoria, sitio):
    df_biciMadf =  pd.read_csv('./data/bici_monu.csv')
    df_biciParkf = pd.read_csv('./data/bicipark_monu.csv')

    print(f'Elegiste: {categoria}')
    print(f'En el lugar: {sitio}')
   #Para el argparse:creamos una lista vacia para meter los nombres ya bien
    sitios_bien = []
    #hacemos que busque el nombre dado en place y lo procesa en forma de tupla del lugar el % aprox del nombre dado
    x = process.extractOne(sitio, df_biciMadf['place'])
    # aqui mete en sitios_bien la parte procesada pero de la primera posión el lugar
    sitios_bien.append(x[0])


    if categoria == 'Bicimad':
        print(f'El bicimad más cercano es: {df_biciMadf[df_biciMadf["place"]==sitios_bien[0]]["station_address"].iloc[0]} ')
        print(f'En esta distancia: {df_biciMadf[df_biciMadf["place"]==sitios_bien[0]]["distance"].iloc[0]} metros')
        #print(df_biciMadf[df_biciMadf['place']==sitio][["place_address", "station_name","station_address", "distance"]])
        #return df_biciMadf[df_biciMadf['place']==sitio]

    if categoria == 'Bicipark':
        print(f'El Bicipark más cercano es: {df_biciParkf[df_biciParkf["place"]==sitios_bien[0]]["station_name"].iloc[0]}')
        print(f'En esta distancia: {df_biciParkf[df_biciParkf["place"]==sitios_bien[0]]["distance"].iloc[0]} metros')
        #print(df_biciParkf[df_biciParkf['place']==sitio][["place_address", "station_name","station_address", "distance"]])

    else:
        return('Elige entre Bicimad y Bicipark')

# Argument parser function
if __name__ == '__main__':
    parser = argparse.ArgumentParser(description= 'Esta es una aplicación para encontrar una estación de bicimad o bicipark dependiendo del Edificios de carácter monumental en el que estes.' )
    help_category ='Elige entre Bicimad o Bicipark'
    help_all ='Aqui te devuelve todo'
    parser.add_argument('-r', help="-r y Elige Bicimad o Bicipark", type=str)
    parser.add_argument('-c', help=help_category, type=str)
    parser.add_argument('-s', help=help_all, type=str)
    args = parser.parse_args()

    if args.r != None:
        resumen(args.r)
    elif args.r == None:
        elegir(args.c, args.s)