python小象学院: datetime----- 52周存钱挑战

""""
    auther:
    function:money challenge
    version:v5.0
    data:2019/3/18
    #record every week saving
    #using :for
    #using:datetime
"""
import math
import datetime


def save_money_in_weeks(money_per_week,increase_money,total_week):
    saving = 0  # Defining global variables
    saved_money_list = []
    # global saving
    money_list = []
    for i in range(total_week):
        money_list.append(money_per_week)
        saving = math.fsum(money_list)
        saved_money_list.append(saving)

        #print 'The {} week,saving {} Yuan,grang total:{}'.format(i + 1, money_per_week, saving)
        money_per_week += increase_money
    return saved_money_list
    # print 'in',saving

def main():
    money_per_week = float(raw_input('Please enter the weekly deposit amount:'))
    increase_money =float(raw_input('Please enter an incremental amount:'))
    total_week  = int(raw_input('Please enter the total number of weeks:'))

    saved_money_list=save_money_in_weeks(money_per_week,increase_money,total_week)

    input_data_str = raw_input('Please enter the data(yyyy/mm/dd):')
    input_data = datetime.datetime.strptime(input_data_str,'%Y/%m/%d')
    week_num = input_data.isocalendar()[1]
    week_num = int(raw_input('Please enter the week which you want to check:'))
    print 'The {} week saved {} Yuan'.format(week_num,saved_money_list[week_num-1])

if __name__ == '__main__':
    main()