-
Notifications
You must be signed in to change notification settings - Fork 0
/
main.go
83 lines (72 loc) · 2.62 KB
/
main.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
package main
import (
"encoding/csv"
"flag"
"log"
"os"
"path"
"github.com/gocarina/gocsv"
)
// const usage = `Usage:
// -v, --verbose verbose output
// -h, --help prints help information
// `
// // for testing, comment out when not needed
// func main() {
// holdings := FetchRecentShareholdings("500209", 28, nil) // nill db will skip db step
// size := fmt.Sprintf("%d", len(holdings.Holdings))
// StructToJson(holdings, "testdata/500209-"+size+"-14012023.json")
// fmt.Printf("%s records written", size)
// }
func main() {
// fmt.Println("##### Marketview v0.1 #####")
var flagFetchSymbols, flagViewSymbols bool
var flagFetchShareholding, flagViewShareHolding string
flag.BoolVar(&flagFetchSymbols, "fetch-sym", false, "Fetch symbols from BSE/NSE Servers")
flag.BoolVar(&flagFetchSymbols, "S", false, "Fetch symbols from BSE/NSE Servers")
flag.BoolVar(&flagViewSymbols, "view-sym", false, "View symbols in csv format")
flag.BoolVar(&flagViewSymbols, "s", false, "View symbols in csv format")
flag.StringVar(&flagFetchShareholding, "fetch-holding", "", "Fetch company's 7 yrs shareholding from BSE Servers")
flag.StringVar(&flagFetchShareholding, "H", "", "Fetch company's 7 yrs shareholding from BSE Servers")
flag.StringVar(&flagViewShareHolding, "view-holding", "", "View Shareholding in csv format")
flag.StringVar(&flagViewShareHolding, "h", "", "View Shareholding in csv format")
// flag.Usage = func() { fmt.Print(usage) }
flag.Parse()
// General Config
_, data_dir := GetConfig()
// symbolsDataDir := path.Join(data_dir, "symbols")
// os.MkdirAll(symbolsDataDir, 0o700)
dbFilepath := path.Join(data_dir, "data.db")
db := GetDB(dbFilepath)
// // ############## Testing code here ##################
// holdings := ViewRecentShareholdingsDb("500209", db)
// log.Print(holdings)
// os.Exit(1)
// // ############# Testing Code Ends ###################
// Controller logic
if flagFetchSymbols {
FetchSymbols(db)
return
} else if flagViewSymbols {
mappings := GetSymbolsMappingDB(db)
err := gocsv.MarshalCSVWithoutHeaders(mappings, csv.NewWriter(os.Stdout))
CheckErr(err)
return
} else if len(flagFetchShareholding) > 0 {
log.Printf("Fetching script id: %s", flagFetchShareholding)
// os.Exit(1)
FetchRecentShareholdings(flagFetchShareholding, 28, db)
return
} else if len(flagViewShareHolding) > 0 {
// log.Println("Not Implemented yet")
// os.Exit(1)
holdings := ViewRecentShareholdingsDb(flagViewShareHolding, db)
// log.Print(holdings)
csv.NewWriter(os.Stdout).WriteAll(holdings)
return
}
// No arguments matched
log.Println("Error! No arguments specified.")
flag.Usage()
os.Exit(1)
}