Untitled

mail@pastecode.io avatarunknown
plain_text
a month ago
2.1 kB
1
Indexable
Never
import string

# Example usage:
# available_singles = ['C1', 'C2']
# available_pairs = [('1A', '2B'), ('2A', '1B')]
# expected result = [('1A', '1B'), ('2A', '2B')]


def check_valid_pair(pair):

    # Unpack the tuple into separate variables
    seat_1, seat_2 = pair

    flag1 = False
    flag2 = False

    # check first number of string is the same
    if seat_1[0] == seat_2[0]:
        flag1 = True

    # check letters are following each other
    alphabet = list(string.ascii_uppercase)
    for i in range(len(alphabet)):
        if seat_1[1] == alphabet[i]:
            if seat_2[1] == alphabet[i+1]:
                flag2 = True

    if flag1 and flag2:
        return True
    else:
        return False

def replace_solo_with_solo_case1(solo1, solo2):
    print("working")

def replace_solo_with_pair_case2(solo1, pair1):
    print("working")

def replace_pair_with_pair_case3(pair1, pair2):
    print("working")
    seat_1, seat_2 = pair1
    seat_3, seat_4 = pair2

    seats_list = [{"old":seat_1, "new": ""}, {"old":seat_2, "new": ""}, {"old":seat_3, "new": ""}, {"old":seat_4, "new": ""}]
    print(seats_list)
    for i in range(len(seats_list)):
        for j in range(len(seats_list)):
            seat_tuple = (seats_list[i]['old'], seats_list[j]['old'])
            if check_valid_pair(seat_tuple):
                print(check_valid_pair(seat_tuple), seat_tuple)
                seats_list[i]["new"] = seats_list[i]['old']
                seats_list[j]["new"] = seats_list[j]['old']
    print(seats_list)

def replace_pair_with_pair_case4(pair1, pair2):
    print("working")



def main():
    # seat_tuple = ('1B', '1G')
    #print(check_valid_pair(seat_tuple))
    # this needs to be per flight and triggerd everytime new pair was added
    available_pairs = [('1A', '2B'), ('2A', '1B')]
    for i in range(len(available_pairs)):
        for j in range(len(available_pairs)):
            replace_pair_with_pair_case3(available_pairs[i], available_pairs[j])


main()