python编程，邮费计算问题，要怎么搞啊

根据邮件的重量和用户是否选择计算邮费。计算规则：重量在1000克以内(包括1000克), 基本费8元。超过1000克的部分，每500克加收超重费4元，不足500克部分按500克计算；如果用户选择加急，多收5元。

只用最基本的语句就好了呀，程序：

weight = int(input('请输入邮件重量（g）：')) #输入重量
if weight > 1000:
     cost = 8 + (((weight - 1000) // 500) + 1) * 4 #超出1000g的部分地板除500，然后不足部分+1
else:
     cost = 8
quick = input('是否加急（yes/no）：')
if quick == 'yes':
     cost += 5 #加急费用+5
print('邮费：',cost,'元')