new price parsing methods + tests
All checks were successful
/ Make image (push) Successful in 39s

This commit is contained in:
nquidox 2026-01-05 17:51:02 +03:00
parent ae0a07ae1b
commit 2fa14a7797
3 changed files with 156 additions and 22 deletions

View file

@ -42,17 +42,19 @@ func (s *Parser) worker(ctx context.Context, receiver chan shared.Task, sender c
//price, err := s.getPrice(pageCtx, task) //price, err := s.getPrice(pageCtx, task)
//pageCancel() //pageCancel()
price, err := s.getMinimalPrice(task) //price, err := s.getMinimalPrice(task)
if err != nil { //if err != nil {
log.WithField("task_uuid", task.MerchUuid).Warn(logHeader + logWorker + logTaskWarning + "failed to process, zero price") // log.WithField("task_uuid", task.MerchUuid).Warn(logHeader + logWorker + logTaskWarning + "failed to process, zero price")
sender <- shared.TaskResult{ // sender <- shared.TaskResult{
MerchUuid: task.MerchUuid, // MerchUuid: task.MerchUuid,
Origin: task.Origin, // Origin: task.Origin,
Price: zeroPrice, // Price: zeroPrice,
} // }
continue // continue
} //}
//price will be zeroPrice value in case of any error or if price not found
price := s.getMinimalPrice(task)
sender <- shared.TaskResult{ sender <- shared.TaskResult{
MerchUuid: task.MerchUuid, MerchUuid: task.MerchUuid,
Origin: task.Origin, Origin: task.Origin,

View file

@ -41,7 +41,7 @@ func (s *Parser) getPrice(ctx context.Context, task shared.Task) (int32, error)
return minimal, nil return minimal, nil
} }
func (s *Parser) getMinimalPrice(task shared.Task) (int32, error) { func (s *Parser) getMinimalPrice(task shared.Task) int32 {
ctx := context.Background() ctx := context.Background()
allocCtx, allocCancel := chromedp.NewRemoteAllocator(ctx, s.externalBrowser) allocCtx, allocCancel := chromedp.NewRemoteAllocator(ctx, s.externalBrowser)
defer allocCancel() defer allocCancel()
@ -52,7 +52,6 @@ func (s *Parser) getMinimalPrice(task shared.Task) (int32, error) {
var ( var (
singlePrice string singlePrice string
rangedPrice string rangedPrice string
prices []int32
) )
if err := chromedp.Run(sessionCtx, if err := chromedp.Run(sessionCtx,
@ -61,26 +60,51 @@ func (s *Parser) getMinimalPrice(task shared.Task) (int32, error) {
chromedp.Evaluate(`(document.querySelector('div.price')?.innerText || '').trim()`, &singlePrice), chromedp.Evaluate(`(document.querySelector('div.price')?.innerText || '').trim()`, &singlePrice),
chromedp.Evaluate(`(document.querySelector('div.price_range')?.innerText || '').trim()`, &rangedPrice), chromedp.Evaluate(`(document.querySelector('div.price_range')?.innerText || '').trim()`, &rangedPrice),
); err != nil { ); err != nil {
return zeroPrice, err return zeroPrice
} }
minimal := s.processPrices(singlePrice, rangedPrice)
log.Infof(logHeader+"uuid: %s, price: %d", task.MerchUuid, minimal)
return minimal
}
func (s *Parser) processPrices(singlePrice, rangedPrice string) int32 {
var prices []int32
//in case of any errors or no price return zeroPrice const
//if success add to prices slice
if singlePrice != "" { if singlePrice != "" {
singlePrice = strings.TrimSpace(singlePrice) singlePrice = strings.TrimSpace(singlePrice)
prices = append(prices, s.getSinglePriceWithTax(singlePrice)) counted, err := s.parseSinglePrice(singlePrice)
if err != nil {
log.WithFields(log.Fields{
"err": err.Error(),
"singlePrice": singlePrice,
}).Error(logHeader + logGetPrice + "failed to parse single price, returning zero price")
return zeroPrice
}
prices = append(prices, counted)
} else { } else {
return zeroPrice, nil //return zero price immediately log.Warn(logHeader + logGetPrice + "single price not found")
return zeroPrice
} }
//optional, adds price only if no errors and has non zero value
if rangedPrice != "" { if rangedPrice != "" {
rangedPrice = strings.TrimSpace(rangedPrice) rangedPrice = strings.TrimSpace(rangedPrice)
prices = append(prices, s.getMinimalPriceFromRangeWithTax(rangedPrice)) counted, err := s.parseRangedPrice(rangedPrice)
//no return because of optional ranged price if err != nil {
log.WithFields(log.Fields{
"err": err.Error(),
"rangedPrice": rangedPrice,
}).Error(logHeader + logGetPrice + "failed to parse ranged price")
} else {
if counted > 0 {
prices = append(prices, counted)
}
}
} }
return slices.Min(prices)
minimal := slices.Min(prices)
log.Infof(logHeader+"uuid: %s, price: %d", task.MerchUuid, minimal)
return minimal, nil
} }
func (s *Parser) getSinglePriceWithTax(rawPrice string) int32 { func (s *Parser) getSinglePriceWithTax(rawPrice string) int32 {
@ -117,3 +141,34 @@ func (s *Parser) getMinimalPriceFromRangeWithTax(priceRange string) int32 {
return int32(float64(price) * taxMultiplier) return int32(float64(price) * taxMultiplier)
} }
func (s *Parser) parseSinglePrice(rawPrice string) (int32, error) {
deCommaStr := strings.ReplaceAll(rawPrice, ",", "")
split := strings.Split(deCommaStr, "円")
finalPrice, err := s.countTax(split[0])
if err != nil {
return zeroPrice, err
}
return finalPrice, nil
}
func (s *Parser) parseRangedPrice(rawPrice string) (int32, error) {
deCommaStr := strings.ReplaceAll(rawPrice, ",", "")
split := strings.Split(deCommaStr, "円")
rm1 := strings.ReplaceAll(split[0], "(", "")
rm2 := strings.ReplaceAll(rm1, "他", "")
finalPrice, err := s.countTax(rm2)
if err != nil {
return zeroPrice, err
}
return finalPrice, nil
}
func (s *Parser) countTax(priceStr string) (int32, error) {
intPrice, err := strconv.Atoi(priceStr)
if err != nil {
return zeroPrice, err
}
return int32(float64(intPrice) * taxMultiplier), nil
}

View file

@ -0,0 +1,77 @@
package mandarake
import (
"context"
"testing"
)
func TestParser_processPrices(t *testing.T) {
type fields struct {
baseCtx context.Context
externalBrowser string
goroutinesNumber int
}
type args struct {
singlePrice string
rangedPrice string
}
var placeholderFields = fields{
baseCtx: context.Background(),
externalBrowser: "",
goroutinesNumber: 10,
}
//single := "18,000円 (税込 19,800円)"
//ranged := "(他15,000円16,000円もあります)"
tests := []struct {
name string
fields fields
args args
want int32
}{
//Cases
{name: "Full success", fields: placeholderFields, args: args{
singlePrice: "18,000円 (税込 19,800円)",
rangedPrice: "(他15,000円16,000円もあります)",
}, want: 16500},
{name: "Single price only success 1", fields: placeholderFields, args: args{
singlePrice: "18,000円 (税込 19,800円)",
rangedPrice: "",
}, want: 19800},
{name: "Single price only success 2", fields: placeholderFields, args: args{
singlePrice: "18,000円 (税込 19,800円)",
rangedPrice: "no numbers in this string",
}, want: 19800},
{name: "zero single price success 1", fields: placeholderFields, args: args{
singlePrice: "",
rangedPrice: "",
}, want: 0},
{name: "zero single price success 2", fields: placeholderFields, args: args{
singlePrice: "no numbers in this string",
rangedPrice: "",
}, want: 0},
{name: "zero single price success 3", fields: placeholderFields, args: args{
singlePrice: "no numbers in this string",
rangedPrice: "no numbers in this string",
}, want: 0},
}
for _, tt := range tests {
t.Run(tt.name, func(t *testing.T) {
s := &Parser{
baseCtx: tt.fields.baseCtx,
externalBrowser: tt.fields.externalBrowser,
goroutinesNumber: tt.fields.goroutinesNumber,
}
if got := s.processPrices(tt.args.singlePrice, tt.args.rangedPrice); got != tt.want {
t.Errorf("processPrices() = %v, want %v", got, tt.want)
}
})
}
}