train.py

#lex_auth_0127136112798105601178    

train_list=[
{"train_no":16453,"name":"Prasanti Express","from":"SBC","to":"BBS","days_of_run":['Mo','We','Th'],"sleeper_fare":600,"ac_fare": 987},
{"train_no":25627,"name":"Karnataka Express","from":"SBC","to":"DEC","days_of_run":['Su','Tu'],"sleeper_fare":1600,"ac_fare": 2500},
{"train_no":22642,"name":"Trivandrum SF Express","from":"VSKP","to":"TVM","days_of_run":['Mo','Tu','We','Th','Fr','Sa'],"sleeper_fare":800,"ac_fare": 1256},
{"train_no":22905,"name":"Okha Howrah Express","from":"ST","to":"KOAA","days_of_run":['We','Sa'],"sleeper_fare":987,"ac_fare": 2879}]

def get_train_name (train_no):
    for x in train_list:
        if train_no == x["train_no"]:
            return x
        
    
    return "Invalid Train_no"
        

def get_trains_for_day(day_of_run):
    li=[]
    for x in train_list:
        if day_of_run in x["days_of_run"]:
            li.append(x["train_no"])
        
    if len(li)==0:
        return "Invalid day"
    else:
        return li
    
   

def get_total_fare(train_no,passenger_dict):
    for x in train_list:
        if train_no == x["train_no"]:
            return( passenger_dict.get("sleeper") * x["sleeper_fare"] + passenger_dict.get("ac") * x["ac_fare"])
            
    
print(get_train_name(25627))
print(get_trains_for_day("Mo"))
print(get_total_fare(22642,{"sleeper":5, "ac":1}))-