将参数从函数传递到主函数

Question

我想知道如何将参数从函数awans（）传递到主函数，以便每年之后staz增加一。这是我现在得到的：

def awans(lata, zarobek, staz):

    for i in range (lata):
        staz += 1
        zarobek = zarobek + 0.1*zarobek
    return zarobek, staz

def drukuj(staz, zarobek):
    print("Po ", staz ," latach stażu bedziesz zarabiał: ", zarobek)

def main(args):
    staz = 1
    zarobek = 1000
    lata = int(input("Podaj przewidywaną ilość lat pracy: "))
    
    zarobek = awans(lata, zarobek, staz)
    drukuj(staz, zarobek)

    return 0

if __name__ == '__main__':
    import sys
    sys.exit(main(sys.argv))