-
Notifications
You must be signed in to change notification settings - Fork 1
/
edgar_collector.go
99 lines (81 loc) · 1.99 KB
/
edgar_collector.go
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
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
package valuator
import (
"errors"
"io"
"log"
"time"
"github.com/palafrank/edgar"
)
type edgarCollector struct {
name string
fetcher edgar.FilingFetcher
store Store
}
func newEdgarCollector(store Store) (Collector, error) {
//Get the Fetcher
fetcher := edgar.NewFilingFetcher()
if fetcher == nil {
return nil, errors.New("Could not create edgar collector")
}
ret := &edgarCollector{
fetcher: fetcher,
store: store,
}
return ret, nil
}
func (c *edgarCollector) MapEdgarFilingToValuatorFiling(fs []edgar.Filing) []Filing {
var ret []Filing
for _, f := range fs {
ret = append(ret, f.(Filing))
}
return ret
}
func (c *edgarCollector) Name() string {
return "Edgar Collector"
}
func (c *edgarCollector) CollectAnnualData(ticker string,
years ...int) ([]Filing, error) {
var err error
var cf edgar.CompanyFolder
var fp io.Reader
if c.store != nil {
fp = c.store.getFinancials(ticker)
}
if fp == nil {
//If there is no historical data. Get it from Edgar.
log.Println("No data found. Fetching from Edgar")
cf, err = c.fetcher.CompanyFolder(ticker, edgar.FilingType10K)
} else {
// If data available in store use that you create folder
cf, err = c.fetcher.CreateFolder(fp, edgar.FilingType10K)
}
if err != nil {
return nil, err
}
// Get all the Available filings
af := cf.AvailableFilings(edgar.FilingType10K)
//Filter out based on what was requested
var filteredAF []time.Time
for _, f := range af {
if len(years) > 0 && !contains(f.Year(), years) {
continue
}
filteredAF = append(filteredAF, f)
}
// Get all the financial data
if len(filteredAF) > 0 {
fils, err := cf.Filings(edgar.FilingType10K, filteredAF...)
if err != nil {
return nil, err
}
return c.MapEdgarFilingToValuatorFiling(fils), nil
}
return nil, errors.New("No filings collected")
}
func (c *edgarCollector) Write(ticker string, writer io.Writer) error {
comp, err := c.fetcher.CompanyFolder(ticker)
if err == nil {
return comp.SaveFolder(writer)
}
return err
}