fix: don't import empty blocks

sentriz · Sep 20, 2021 · 163fd46 · 163fd46
1 parent 466a9df
commit 163fd46
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 0 deletions.
diff --git a/backend/db/db.go b/backend/db/db.go
@@ -219,6 +219,10 @@ func (db *DB) SearchMedias(options SearchMediasOptions) ([]*Media, error) {
 }
 
 func (db *DB) CreateBlocks(blocks []*Block) error {
+	if len(blocks) == 0 {
+		return nil
+	}
+
 	q := db.
 		Insert("blocks").
 		Columns("media_id", "index", "min_x", "min_y", "max_x", "max_y", "body")

diff --git a/backend/importer/importer.go b/backend/importer/importer.go
@@ -263,6 +263,10 @@ func (i *Importer) insertBlocks(id db.MediaID, image image.Image) error {
 
 	blocks := make([]*db.Block, 0, len(rawBlocks))
 	for idx, rawBlock := range rawBlocks {
+		if strings.TrimSpace(rawBlock.Word) == "" {
+			continue
+		}
+
 		rect := imagery.ScaleDownRect(rawBlock.Box)
 		blocks = append(blocks, &db.Block{
 			MediaID: id,
@@ -274,6 +278,7 @@ func (i *Importer) insertBlocks(id db.MediaID, image image.Image) error {
 			Body:    rawBlock.Word,
 		})
 	}
+
 	if err := i.db.CreateBlocks(blocks); err != nil {
 		return fmt.Errorf("inserting blocks: %w", err)
 	}