diff options
Diffstat (limited to 'src/main.go')
| -rw-r--r-- | src/main.go | 32 |
1 files changed, 21 insertions, 11 deletions
diff --git a/src/main.go b/src/main.go index 272e4b6..cf1c74b 100644 --- a/src/main.go +++ b/src/main.go @@ -1,6 +1,7 @@ package main import ( + "flag" "fmt" "log" "regexp" @@ -164,6 +165,9 @@ func (a *App) extractImdbIDs() error { } func main() { + wikiOnly := flag.Bool("wiki-only", false, "only fetch wiki article data, skip SPARQL") + flag.Parse() + cfg, err := LoadConfig("config.json") if err != nil { log.Fatalf("failed to load config: %v", err) @@ -183,19 +187,25 @@ func main() { } defer app.DB.Close() - if err = app.extractImdbIDs(); err != nil { - log.Fatalf("extractImdbIDs: %v", err) - } + if !*wikiOnly { + if err = app.extractImdbIDs(); err != nil { + log.Fatalf("extractImdbIDs: %v", err) + } - if err = app.populateImdbTable(); err != nil { - log.Fatalf("populateImdbTable: %v", err) - } + if err = app.populateImdbTable(); err != nil { + log.Fatalf("populateImdbTable: %v", err) + } - if err = app.fetchAndUpdateImdbData(); err != nil { - log.Fatalf("fetchAndUpdateImdbData: %v", err) - } + if err = app.fetchAndUpdateImdbData(); err != nil { + log.Fatalf("fetchAndUpdateImdbData: %v", err) + } - if err = app.fetchWikiArticles(); err != nil { - log.Fatalf("fetchWikiArticles: %v", err) + if err = app.fetchWikiArticles(); err != nil { + log.Fatalf("fetchWikiArticles: %v", err) + } + } else { + if err = app.fetchWikiArticlesData(); err != nil { + log.Fatalf("fetchWikiArticlesData: %v", err) + } } } |