37 lines
783 B
Go
37 lines
783 B
Go
package cmd
|
|
|
|
import (
|
|
"github.com/spf13/cobra"
|
|
"github.com/vilmibm/trunkless/db"
|
|
"github.com/vilmibm/trunkless/ingest"
|
|
)
|
|
|
|
func init() {
|
|
ingestCmd.Flags().StringP("cutupdir", "d", "", "directory to files produced by cutup cmd")
|
|
ingestCmd.MarkFlagRequired("cutupdir")
|
|
rootCmd.AddCommand(ingestCmd)
|
|
}
|
|
|
|
var ingestCmd = &cobra.Command{
|
|
Use: "ingest corpusname",
|
|
Short: "ingest already cut-up corpora from disk into database",
|
|
Args: cobra.ExactArgs(1),
|
|
RunE: func(cmd *cobra.Command, args []string) error {
|
|
cutupDir, _ := cmd.Flags().GetString("cutupdir")
|
|
corpus := args[0]
|
|
|
|
conn, err := db.Connect()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
opts := ingest.IngestOpts{
|
|
Conn: conn,
|
|
CutupDir: cutupDir,
|
|
Corpus: corpus,
|
|
}
|
|
|
|
return ingest.Ingest(opts)
|
|
},
|
|
}
|