import pandas


# Functions go here
def make_statement(statement, decoration, lines=1):
    """Creates headings (3 lines), subheadings (2lines) and emphasised text
     / mini-headings (1 line). Only use emoji for single line statements"""

    middle = f"{decoration * 3} {statement} {decoration * 3}"
    top_bottom = decoration * len(middle)

    if lines == 1:
        print(middle)
    elif lines == 2:
        print(middle)
        print(top_bottom)

    else:
        print(top_bottom)
        print(middle)
        print(top_bottom)


# currency formatting function
def currency(x):
    """Format numbers as currency ($#.##)"""
    return "${:.2f}".format(x)


# burger lists 1
burger_items = ["Cheeseburger", "Bacon Burger", "Chicken Burger", "Veggie Burger", "Double Beef",
                "BBQ Burger", "Mushroom Swiss", "Fish Burger", "Spicy Jalapeño", "Classic Burger"]
burger_cost = [7.50] * 10
burger_dict_1 = {
    'Burger': burger_items,
    'Price': burger_cost,
}
# burger list 2
special_burger = ["Lamb Burger", "Pulled Pork Burger"]
special_burger_cost = [8.50, 9.50]
burger_dict_2 = {
    'Special Burger': special_burger,
    'Price': special_burger_cost,
}
# burger sizes
burger_size = ["Small", "Medium", "Large"]
burger_size_cost = [2.50, 3.50, 4.50]
burger_dict_3 = {
    'Sizes': burger_size,
    'Price': burger_size_cost,
}
# create dataframe/s
burger_frame = pandas.DataFrame(burger_dict_1)
burger_frame.index = burger_frame.index + 1

burger_frame_2 = pandas.DataFrame(burger_dict_2)
burger_frame_2.index = burger_frame_2.index + 11

burger_frame_3 = pandas.DataFrame(burger_dict_3)
burger_frame_3.index = burger_frame_3.index + 1

# Currency Formatting (uses currency function)
add_dollars = ['Price']
for var_item in add_dollars:
    burger_frame[var_item] = burger_frame[var_item].apply(currency)
    burger_frame_2[var_item] = burger_frame_2[var_item].apply(currency)
    burger_frame_3[var_item] = burger_frame_3[var_item].apply(currency)

# Main routine goes here
make_statement(statement="Bun Bro's", decoration="🍔")
print(burger_frame)
make_statement(statement="", decoration="----")
print(burger_frame_2)
make_statement(statement="", decoration="----")
print(burger_frame_3)