Handle error when single tweet dosen't exist

mmpx12 · Mar 10, 2024 · 73e9f12 · 73e9f12
1 parent 06c58e1
commit 73e9f12
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/twmd.go b/twmd.go
@@ -156,6 +156,9 @@ func photoUser(wait *sync.WaitGroup, tweet *twitterscraper.TweetResult, output s
 }
 
 func videoSingle(tweet *twitterscraper.Tweet, output string) {
+	if tweet == nil {
+		return
+	}
 	if len(tweet.Videos) > 0 {
 		wg := sync.WaitGroup{}
 		for _, i := range tweet.Videos {
@@ -174,6 +177,9 @@ func videoSingle(tweet *twitterscraper.Tweet, output string) {
 }
 
 func photoSingle(tweet *twitterscraper.Tweet, output string) {
+	if tweet == nil {
+		return
+	}
 	if len(tweet.Photos) > 0 {
 		wg := sync.WaitGroup{}
 		for _, i := range tweet.Photos {
@@ -250,6 +256,10 @@ func singleTweet(output string, id string) {
 		fmt.Println(err)
 		os.Exit(1)
 	}
+	if tweet == nil {
+		fmt.Println("Error retrieve tweet")
+		return
+	}
 	if usr != "" {
 		if vidz {
 			videoSingle(tweet, output)