this repo has no description
labeling: rename kw labeler files; make methods public
+11
-11
+7
-7
labeling/labelers.go
labeling/keyword_labeler.go
+7
-7
labeling/labelers.go
labeling/keyword_labeler.go
···
8
8
9
9
// simple record labeling (without pre-fetched blobs)
10
10
type SimplePostLabeler interface {
11
-
labelPost(p bsky.FeedPost) []string
11
+
LabelPost(p bsky.FeedPost) []string
12
12
}
13
13
type SimpleActorProfileLabeler interface {
14
-
labelActorProfile(ap bsky.ActorProfile) []string
14
+
LabelActorProfile(ap bsky.ActorProfile) []string
15
15
}
16
16
17
17
type KeywordLabeler struct {
···
19
19
value string
20
20
}
21
21
22
-
func (kl KeywordLabeler) labelText(txt string) []string {
22
+
func (kl KeywordLabeler) LabelText(txt string) []string {
23
23
txt = strings.ToLower(txt)
24
24
for _, word := range kl.keywords {
25
25
if strings.Contains(txt, word) {
···
29
29
return []string{}
30
30
}
31
31
32
-
func (kl KeywordLabeler) labelPost(p bsky.FeedPost) []string {
33
-
return kl.labelText(p.Text)
32
+
func (kl KeywordLabeler) LabelPost(p bsky.FeedPost) []string {
33
+
return kl.LabelText(p.Text)
34
34
}
35
35
36
-
func (kl KeywordLabeler) labelActorProfile(ap bsky.ActorProfile) []string {
36
+
func (kl KeywordLabeler) LabelActorProfile(ap bsky.ActorProfile) []string {
37
37
txt := ap.DisplayName
38
38
if ap.Description != nil {
39
39
txt += *ap.Description
40
40
}
41
-
return kl.labelText(txt)
41
+
return kl.LabelText(txt)
42
42
}
+2
-2
labeling/labelers_test.go
labeling/keyword_labeler_test.go
+2
-2
labeling/labelers_test.go
labeling/keyword_labeler_test.go
···
21
21
}
22
22
23
23
for _, c := range postCases {
24
-
vals := kl.labelPost(c.record)
24
+
vals := kl.LabelPost(c.record)
25
25
if !reflect.DeepEqual(vals, c.expected) {
26
26
t.Log(fmt.Sprintf("labels expected:%s got:%s", c.expected, vals))
27
27
t.Fail()
···
41
41
}
42
42
43
43
for _, c := range profileCases {
44
-
vals := kl.labelActorProfile(c.record)
44
+
vals := kl.LabelActorProfile(c.record)
45
45
if !reflect.DeepEqual(vals, c.expected) {
46
46
t.Log(fmt.Sprintf("labels expected:%s got:%s", c.expected, vals))
47
47
t.Fail()
+2
-2
labeling/service.go
+2
-2
labeling/service.go
···
140
140
}
141
141
// run through all the keyword labelers on posts, saving any resulting labels
142
142
for _, labeler := range s.kwl {
143
-
for _, val := range labeler.labelPost(*post) {
143
+
for _, val := range labeler.LabelPost(*post) {
144
144
labelVals = append(labelVals, val)
145
145
}
146
146
}
···
151
151
}
152
152
// run through all the keyword labelers on posts, saving any resulting labels
153
153
for _, labeler := range s.kwl {
154
-
for _, val := range labeler.labelActorProfile(*profile) {
154
+
for _, val := range labeler.LabelActorProfile(*profile) {
155
155
labelVals = append(labelVals, val)
156
156
}
157
157
}