Get Indian MF prices

hledger_get-indian-mf-prices.py

import subprocess
import datetime
import urllib.request 

def interesting_commodity(commodities, line):
    """Returns True if the given line has one of the given commodities"""
    for c in commodities:
        if c in line:
            return True
    return False


def main():
    with open("rates.journal", "a") as rates:
        data = urllib.request.urlopen("https://www.amfiindia.com/spages/NAVAll.txt")
        commodities = set(x.strip() for x in open("./hledger-commodities"))
        for idx,i in enumerate(data):
            if idx == 40:
                break
            i = i.decode('utf-8')
            if interesting_commodity(commodities, i):
                fields = [x.strip() for x in i.split(';')]
                scheme_code, div_payout, div_reinvestment, scheme, net_value, repurchase, sale, date = fields
                date = datetime.datetime.strptime("25-May-2018", "%d-%b-%Y")
                rates.write('"{}" ₹{}\n'.format(div_reinvestment, repurchase))
                rates.write("P {} {} {}\n".format(scheme, div_payout, div_reinvestment))
                
           
if __name__ == "__main__":
    main()