feat: added low level API for Database exports
Signed-off-by: SoXX <soxx@fenpa.ws>
This commit is contained in:
parent
e2bce2bda7
commit
6124a3fc61
51
example/lowlevel/dbexport.go
Normal file
51
example/lowlevel/dbexport.go
Normal file
@ -0,0 +1,51 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"git.dragse.it/anthrove/e621-sdk-go/pkg/e621/endpoints"
|
||||
"git.dragse.it/anthrove/e621-sdk-go/pkg/e621/model"
|
||||
"io"
|
||||
"log"
|
||||
"net/http"
|
||||
"os"
|
||||
)
|
||||
|
||||
func main() {
|
||||
// Define the request context with essential information.
|
||||
requestContext := model.RequestContext{
|
||||
Client: http.Client{},
|
||||
Host: "https://e621.net",
|
||||
UserAgent: "Go-e621-SDK (@username)",
|
||||
Username: os.Getenv("API_USER"), // Replace with your username
|
||||
APIKey: os.Getenv("API_KEY"), // Replace with your API key
|
||||
}
|
||||
|
||||
log.Println("Getting a list of DB Exports: ")
|
||||
dbExportFiles, err := endpoints.GetDBExportList(requestContext)
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
|
||||
log.Printf("%d files found", len(dbExportFiles))
|
||||
for _, v := range dbExportFiles {
|
||||
log.Printf("File found: %s", v)
|
||||
}
|
||||
|
||||
exportFileName := dbExportFiles[0]
|
||||
log.Println("Downloading DB export")
|
||||
log.Printf("File to download: %s", exportFileName)
|
||||
rawFile, err := endpoints.GetDBExportFile(requestContext, exportFileName)
|
||||
|
||||
file, err := os.Create(exportFileName)
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
|
||||
defer file.Close()
|
||||
|
||||
_, err = io.Copy(file, rawFile)
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
log.Printf("File %s downloaded", exportFileName)
|
||||
|
||||
}
|
@ -5,6 +5,7 @@ import (
|
||||
"git.dragse.it/anthrove/e621-sdk-go/pkg/e621/model"
|
||||
"git.dragse.it/anthrove/e621-sdk-go/pkg/e621/utils"
|
||||
"golang.org/x/net/html"
|
||||
"io"
|
||||
"net/http"
|
||||
"strings"
|
||||
)
|
||||
@ -83,9 +84,9 @@ func GetDBExportList(requestContext model.RequestContext) ([]string, error) {
|
||||
// - file: The name of the file to be fetched from the database export.
|
||||
//
|
||||
// Returns:
|
||||
// - *http.Response: The HTTP response containing the requested file (probably a csv.gz).
|
||||
// - io.ReadCloser: The HTTP response containing the requested file (probably a csv.gz).
|
||||
// - error: An error, if any, encountered during the API request or response handling.
|
||||
func GetDBExportFile(requestContext model.RequestContext, file string) (*http.Response, error) {
|
||||
func GetDBExportFile(requestContext model.RequestContext, file string) (io.ReadCloser, error) {
|
||||
if file == "" {
|
||||
return nil, fmt.Errorf("no file specified")
|
||||
}
|
||||
@ -111,7 +112,6 @@ func GetDBExportFile(requestContext model.RequestContext, file string) (*http.Re
|
||||
// If the status code is outside the 2xx range, return an error based on the status code.
|
||||
return nil, utils.StatusCodesToError(resp.StatusCode)
|
||||
}
|
||||
|
||||
return resp, nil
|
||||
return resp.Body, nil
|
||||
|
||||
}
|
||||
|
Reference in New Issue
Block a user