-
Notifications
You must be signed in to change notification settings - Fork 0
/
finance.py
executable file
·77 lines (57 loc) · 2.73 KB
/
finance.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
#!/usr/bin/python
from FinanceChart import *
# Create a finance chart demo containing 100 days of data
noOfDays = 100
# To compute moving averages starting from the first day, we need to get extra data
# points before the first day
extraDays = 30
# In this exammple, we use a random number generator utility to simulate the data. We
# set up the random table to create 6 cols x (noOfDays + extraDays) rows, using 9 as
# the seed.
rantable = RanTable(9, 6, noOfDays + extraDays)
# Set the 1st col to be the timeStamp, starting from Sep 4, 2002, with each row
# representing one day, and counting week days only (jump over Sat and Sun)
rantable.setDateCol(0, chartTime(2002, 9, 4), 86400, 1)
# Set the 2nd, 3rd, 4th and 5th columns to be high, low, open and close data. The
# open value starts from 100, and the daily change is random from -5 to 5.
rantable.setHLOCCols(1, 100, -5, 5)
# Set the 6th column as the vol data from 5 to 25 million
rantable.setCol(5, 50000000, 250000000)
# Now we read the data from the table into arrays
timeStamps = rantable.getCol(0)
highData = rantable.getCol(1)
lowData = rantable.getCol(2)
openData = rantable.getCol(3)
closeData = rantable.getCol(4)
volData = rantable.getCol(5)
# Create a FinanceChart object of width 640 pixels
c = FinanceChart(640)
# Add a title to the chart
c.addTitle("Finance Chart Demonstration")
# Set the data into the finance chart object
c.setData(timeStamps, highData, lowData, openData, closeData, volData, extraDays)
# Add the main chart with 240 pixels in height
c.addMainChart(240)
# Add a 5 period simple moving average to the main chart, using brown color
c.addSimpleMovingAvg(5, '0x663300')
# Add a 20 period simple moving average to the main chart, using purple color
c.addSimpleMovingAvg(20, '0x9900ff')
# Add HLOC symbols to the main chart, using green/red for up/down days
c.addHLOC('0x008000', '0xcc0000')
# Add 20 days bollinger band to the main chart, using light blue (9999ff) as the
# border and semi-transparent blue (c06666ff) as the fill color
c.addBollingerBand(20, 2, '0x9999ff', '0xc06666ff')
# Add a 75 pixels volume bars sub-chart to the bottom of the main chart, using
# green/red/grey for up/down/flat days
c.addVolBars(75, '0x99ff99', '0xff9999', '0x808080')
# Append a 14-days RSI indicator chart (75 pixels high) after the main chart. The
# main RSI line is purple (800080). Set threshold region to +/- 20 (that is, RSI = 50
# +/- 25). The upper/lower threshold regions will be filled with red (ff0000)/blue
# (0000ff).
c.addRSI(75, 14, '0x800080', 20, '0xff0000', '0x0000ff')
# Append a 12-days momentum indicator chart (75 pixels high) using blue (0000ff)
# color.
c.addMomentum(75, 12, '0x0000ff')
# Output the chart
print("Content-type: image/png\n")
binaryPrint(c.makeChart2(PNG))