anthrove/e621-sdk-go
Archived
4
0
Fork 0
Code Issues 3 Pull Requests Packages Projects Releases Wiki Activity

feat: added getting all posts % some fixes

Browse Source 
Signed-off-by: SoXX <soxx@fenpa.ws>
This commit is contained in:
SoXX 2023-11-06 13:46:54 +01:00
parent f446823abd
commit 4c2677dc58
3 changed files with 29 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
example/highlevel/post.go
View File

@ -29,7 +29,14 @@ func main() {
log.Printf("Post 1337 has following tags: %s", post.Tags) log.Printf("Post 1337 has following tags: %s", post.Tags)
postBuilder := client.GetPosts() postBuilder := client.GetPosts()
posts, err = client.GetNPosts(230, postBuilder) posts1, err := client.GetNPosts(600, postBuilder)
if err != nil {
log.Panic(err)
}
log.Println(len(posts1))
postBuilder = client.GetPosts().Tags("how_to_dragon_your_train")
posts, err = client.GetAllPosts(postBuilder)
if err != nil { if err != nil {
log.Panic(err) log.Panic(err)
} }

10
pkg/e621/builder/posts.go
View File

@ -11,7 +11,8 @@ import (
type PostsBuilder interface { type PostsBuilder interface {
Tags(tags string) PostsBuilder Tags(tags string) PostsBuilder
PageAfter(postID model.PostID) PostsBuilder PageAfter(postID model.PostID) PostsBuilder
pageBefore(postID model.PostID) PostsBuilder PageBefore(postID model.PostID) PostsBuilder
PageNumber(number int) PostsBuilder
SetLimit(limitUser int) PostsBuilder SetLimit(limitUser int) PostsBuilder
Execute() ([]model.Post, error) Execute() ([]model.Post, error)
} }
@ -40,11 +41,16 @@ func (g *getPosts) PageAfter(postID model.PostID) PostsBuilder {
return g return g
} }
func (g *getPosts) pageBefore(postID model.PostID) PostsBuilder { func (g *getPosts) PageBefore(postID model.PostID) PostsBuilder {
g.query["page"] = "b" + strconv.Itoa(int(postID)) g.query["page"] = "b" + strconv.Itoa(int(postID))
return g return g
} }
func (g *getPosts) PageNumber(number int) PostsBuilder {
g.query["page"] = strconv.Itoa(number)
return g
}
func (g *getPosts) SetLimit(limitUser int) PostsBuilder { func (g *getPosts) SetLimit(limitUser int) PostsBuilder {
g.query["limit"] = strconv.Itoa(limitUser) g.query["limit"] = strconv.Itoa(limitUser)
return g return g

14
pkg/e621/client.go
View File

@ -7,6 +7,8 @@ import (
"git.dragse.it/anthrove/e621-sdk-go/pkg/e621/utils" "git.dragse.it/anthrove/e621-sdk-go/pkg/e621/utils"
_ "github.com/joho/godotenv/autoload" _ "github.com/joho/godotenv/autoload"
"golang.org/x/time/rate" "golang.org/x/time/rate"
"log"
"math"
"net/http" "net/http"
"strconv" "strconv"
) )
@ -79,13 +81,23 @@ func (c *Client) GetNPosts(n int, postBuilder builder.PostsBuilder) ([]model.Pos
} }
for len(posts) < n { for len(posts) < n {
postBuilder.PageAfter(posts[len(posts)-1].ID).SetLimit(n - len(posts)) lastPostID := posts[len(posts)-1].ID
postBuilder.PageBefore(lastPostID)
postBuilder.SetLimit(n - len(posts))
newPosts, err := postBuilder.Execute() newPosts, err := postBuilder.Execute()
if err != nil { if err != nil {
return nil, err return nil, err
} }
if len(newPosts) == 0 {
break
}
posts = append(posts, newPosts...) posts = append(posts, newPosts...)
log.Printf("Post ID: %d | Post Size: %d", lastPostID, len(posts))
} }
return posts, nil return posts, nil
} }
func (c *Client) GetAllPosts(postBuilder builder.PostsBuilder) ([]model.Post, error) {
return c.GetNPosts(math.MaxInt, postBuilder)
}