from SQL import SQL

class Client(SQL):

    def __init__(self,HOST=None, DATABASE=None, PASSWORD=None, USER=None):
        super().__init__(HOST, DATABASE, PASSWORD, USER)

    # TO DO
    def consulter_horaire_train(self,numero):
        sql='''
            SELECT depart, arrivee
            FROM horaires INNER JOIN train
            ON train.numero = horaires.id_train
            WHERE train.numero=%d;
            ''' % numero
        return self._requete_sql(sql)

    # TO DO
    def rechercher_trajet(self, ville_depart, ville_arrivee):
        sql='''
            SELECT numero
            FROM (train INNER JOIN horaires
            ON train.numero = horaires.id_train)
            INNER JOIN arret
            ON arret.gare = gare.id_gare
            WHERE (gare.nom_ville='%s' OR gare.nom_ville='%s');
            ''' % (ville_depart,ville_arrivee)
        return self._requete_sql(sql)