feat: added low level API for Database exports

Signed-off-by: SoXX <soxx@fenpa.ws>
This commit is contained in:
SoXX 2023-11-13 11:45:39 +01:00
parent e2bce2bda7
commit 6124a3fc61
2 changed files with 55 additions and 4 deletions

View 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)
}

View File

@ -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
}