hitung_barang.py with discount

"""
buatlah program sederhana
dihitung pake for 
user memasukkan harga barang
user memasukkan total barang
lalu print total harga barang
bisa pakai operator +=
untuk barang ke -3 di kasih discount 50%
contoh: outputnya 
total barang: 7 
harga barang: 10000
barang ke  1 total_harga=  10000
barang ke  2 total_harga=  20000
harga barang ke  3 = 5000.0
barang ke  3 total_harga=  25000.0
barang ke  3 total_harga=  25000.0
barang ke  4 total_harga=  35000.0
barang ke  5 total_harga=  45000.0
barang ke  6 total_harga=  55000.0
barang ke  7 total_harga=  65000.0
65000.0
"""


barang = int(input('total barang: '))
harga_barang = int(input('harga barang: '))
total_harga= 0

for i in range(barang):

    if i+1==3:
        total_harga+= (harga_barang*0.5)
        print('harga barang ke 3',"=", (harga_barang*0.5))
        print('barang ke ',i+1,"total_harga= ",total_harga )
    else :
           total_harga += harga_barang
    print('barang ke ',i+1,"total_harga= ",total_harga )

print(total_harga)