52 lines
1.2 KiB
Go
52 lines
1.2 KiB
Go
|
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)
|
||
|
|
|
||
|
|
}
|