mirror of
https://github.com/Gnucash/gnucash.git
synced 2025-02-25 18:55:30 -06:00
83d14e1c1c
It is split into - /libgnucash (for the non-gui bits) - /gnucash (for the gui) - /common (misc source files used by both) - /bindings (currently only holds python bindings) This is the first step in restructuring the code. It will need much more fine tuning later on.
93 lines
2.4 KiB
Python
93 lines
2.4 KiB
Python
#!/usr/bin/env python
|
|
|
|
# quotes_historic.py -- Example Script to read historic quote data into gnucash
|
|
#
|
|
|
|
## @file
|
|
# @brief Example Script to read historic stock data into gnucash
|
|
# @author Peter Holtermann
|
|
# @date January 2011
|
|
# @ingroup python_bindings_examples
|
|
#
|
|
# Call the perl-script @code
|
|
# ./get_quotes.pl INTC
|
|
# @endcode first to achieve data into file INTC which can thereafter be imported to GnuCash using this script.
|
|
#
|
|
# For explanation of use have a look at the wiki:
|
|
# http://wiki.gnucash.org/wiki/Stocks/get_prices
|
|
#
|
|
|
|
from gnucash import Session, Account, Split
|
|
import gnucash
|
|
import datetime
|
|
from fractions import Fraction
|
|
from gnc_convenience import find_account
|
|
|
|
FILE = "./test.gnucash"
|
|
url = "xml://"+FILE
|
|
|
|
# Read data from file
|
|
f = open('INTC')
|
|
data = []
|
|
while 1:
|
|
tmp = f.readline()
|
|
if(len(tmp)<2):
|
|
break
|
|
|
|
data.append(tmp)
|
|
|
|
f.close()
|
|
|
|
stock_date = []
|
|
stock_price = []
|
|
for i in range(1,len(data)):
|
|
year = int(data[i].rsplit(',')[1].rsplit('/')[0])
|
|
month = int(data[i].rsplit(',')[1].rsplit('/')[1])
|
|
day = int(data[i].rsplit(',')[1].rsplit('/')[2])
|
|
stock_date.append(datetime.datetime(year,month,day))
|
|
stock_price.append(float(data[i].rsplit(',')[5]))
|
|
|
|
# Initialize Gnucash session
|
|
session = Session(url, True, False, False)
|
|
root = session.book.get_root_account()
|
|
book = session.book
|
|
account = book.get_root_account()
|
|
pdb = book.get_price_db()
|
|
comm_table = book.get_table()
|
|
ac = find_account(account,'Intel')[0]
|
|
|
|
stock = ac.GetCommodity()
|
|
# Add the prices
|
|
pdb = book.get_price_db()
|
|
if len(ac.GetSplitList())<1:
|
|
print 'Need at least one Split to get currency info ... '
|
|
raise SystemExit
|
|
cur = ac.GetSplitList()[0].GetParent().GetCurrency()
|
|
|
|
# Get stock data
|
|
pl = pdb.get_prices(stock,cur)
|
|
if len(pl)<1:
|
|
print 'Need at least one database entry to clone ...'
|
|
raise SystemExit
|
|
|
|
pl0 = pl[0]
|
|
for i in range(1,len(pl)):
|
|
pdb.remove_price(pl[i])
|
|
|
|
for i in range(0,len(stock_date)):
|
|
p_new = pl0.clone(book)
|
|
p_new = gnucash.GncPrice(instance=p_new)
|
|
print 'Adding',i,stock_date[i],stock_price[i]
|
|
p_new.set_time(stock_date[i])
|
|
v = p_new.get_value()
|
|
v.num = int(Fraction.from_float(stock_price[i]).limit_denominator(100000).numerator)
|
|
v.denom = int(Fraction.from_float(stock_price[i]).limit_denominator(100000).denominator)
|
|
p_new.set_value(v)
|
|
p_new.set_source("Finance::Quotes::Historic")
|
|
pdb.add_price(p_new)
|
|
|
|
# Clean up
|
|
session.save()
|
|
session.end()
|
|
session.destroy()
|