Add update Like flag

JustHumanz · Apr 29, 2022 · 882704c · 882704c
1 parent 72643ec
commit 882704c
Show file tree

Hide file tree

Showing 3 changed files with 14 additions and 13 deletions.
diff --git a/pkg/database/fanart.go b/pkg/database/fanart.go
@@ -214,7 +214,7 @@ func (Data DataFanart) AddLewd() error {
 }
 
 //CheckMemberFanart Check if `that` was a new fanart
-func (FanArt DataFanart) CheckTweetFanArt() (bool, error) {
+func (FanArt DataFanart) CheckTweetFanArt(Update bool) (bool, error) {
 	if FanArt.Lewd {
 		var (
 			id int
@@ -258,7 +258,7 @@ func (FanArt DataFanart) CheckTweetFanArt() (bool, error) {
 			return true, nil
 		} else if err != nil {
 			return false, err
-		} else {
+		} else if Update {
 			//update like
 			log.WithFields(log.Fields{
 				"Name":    FanArt.Member.EnName,
@@ -488,7 +488,7 @@ func (p *Member) GetRandomLewd() (*DataFanart, error) {
 	return b, nil
 }
 
-func (Member Member) ScrapTwitterFanart(Scraper *twitterscraper.Scraper, Lewd bool) ([]DataFanart, error) {
+func (Member Member) ScrapTwitterFanart(Scraper *twitterscraper.Scraper, Lewd bool, update bool) ([]DataFanart, error) {
 
 	var (
 		FanartList []DataFanart
@@ -535,7 +535,7 @@ func (Member Member) ScrapTwitterFanart(Scraper *twitterscraper.Scraper, Lewd bo
 					TweetArt.Videos = tweet.Videos[0].Preview
 				}
 
-				New, err := TweetArt.CheckTweetFanArt()
+				New, err := TweetArt.CheckTweetFanArt(update)
 				if err != nil {
 					return nil, err
 				}

diff --git a/service/fanart/twitter/twitter.go b/service/fanart/twitter/twitter.go
@@ -24,6 +24,7 @@ var (
 	Bot          *discordgo.Session
 	lewd         = flag.Bool("LewdFanart", false, "Enable lewd fanart module")
 	torTransport = flag.Bool("Tor", false, "Enable multiTor for bot transport")
+	like         = flag.Bool("Like", false, "Update like fanart")
 	gRCPconn     pilot.PilotServiceClient
 	ServiceUUID  = uuid.New().String()
 )
@@ -155,12 +156,12 @@ func (i *checkTwJob) Run() {
 			"Lewd":    false,
 		}).Info("Start curl twitter")
 
-		Fanarts, err := Member.ScrapTwitterFanart(engine.InitTwitterScraper(), false)
+		Fanarts, err := Member.ScrapTwitterFanart(engine.InitTwitterScraper(), false, *like)
 		if err != nil {
 			log.WithFields(log.Fields{
-				"vtuber":  Member.Name,
-				"agency":  Member.Group.GroupName,
-				"hashtag": Member.TwitterHashtag,
+				"Vtuber":  Member.Name,
+				"Agency":  Member.Group.GroupName,
+				"Hashtag": Member.TwitterHashtag,
 			}).Error(err)
 			return
 		}
@@ -183,12 +184,12 @@ func (i *checkTwJob) Run() {
 				"Lewd":    true,
 			}).Info("Start curl twitter")
 
-			Fanarts, err := Member.ScrapTwitterFanart(engine.InitTwitterScraper(), true)
+			Fanarts, err := Member.ScrapTwitterFanart(engine.InitTwitterScraper(), true, *like)
 			if err != nil {
 				log.WithFields(log.Fields{
-					"vtuber":  Member.Name,
-					"agency":  Member.Group.GroupName,
-					"hashtag": Member.TwitterLewd,
+					"Vtuber":  Member.Name,
+					"Agency":  Member.Group.GroupName,
+					"Hashtag": Member.TwitterLewd,
 				}).Error(err)
 				return
 			}

diff --git a/service/migrate/scraper.go b/service/migrate/scraper.go
@@ -29,7 +29,7 @@ func TwitterFanart() {
 	}
 	for _, Group := range Groups {
 		for _, Member := range Group.Members {
-			_, err := Member.ScrapTwitterFanart(scraper, false)
+			_, err := Member.ScrapTwitterFanart(scraper, false, true)
 			if err != nil {
 				log.WithFields(log.Fields{
 					"Vtuber": Member.Name,