/waifuVault-python-api

Official Python SDK for waifuVault

Primary LanguagePythonMIT LicenseMIT

waifuvault-python-api

tests GitHub issues last-commit

This contains the official API bindings for uploading, deleting and obtaining files with waifuvault.moe. Contains a full up to date API for interacting with the service

Installation

pip install waifuvault

Usage

This API contains 10 interactions:

  1. Upload File
  2. Get file Info
  3. Update file Info
  4. Delete File
  5. Get File
  6. Create Bucket
  7. Delete Bucket
  8. Get Bucket
  9. Get Restrictions
  10. Clear Restrictions

The package is namespaced to waifuvault, so to import it, simply:

import waifuvault

Upload File

To Upload a file, use the upload_file function. This function takes the following options as an object:

Option Type Description Required Extra info
target string or buffer The target to upload can be a buffer, URL or filename true URL or file path
target_name string The filename of the target if it is a buffer true if buffer Filename with extension
bucket_token string Token for a bucket to upload the file into false Create bucket gives token
expires string A string containing a number and a unit (1d = 1day) false Valid units are m, h and d
hideFilename boolean If true, then the uploaded filename won't appear in the URL false Defaults to false
password string If set, then the uploaded file will be encrypted false
oneTimeDownload boolean if supplied, the file will be deleted as soon as it is accessed false

NOTE: Server restrictions are checked by the SDK client side before upload, and will throw a ValueError exception if they are violated

Using a URL:

import waifuvault
upload_file = waifuvault.FileUpload("https://waifuvault.moe/assets/custom/images/08.png")
upload_res = waifuvault.upload_file(upload_file)
print(f"{upload_res.url}")

Using a file path:

import waifuvault
upload_file = waifuvault.FileUpload("./files/aCoolFile.png")
upload_res = waifuvault.upload_file(upload_file)
print(f"{upload_res.url}")

Using a buffer:

import waifuvault
import io

with open("./files/aCoolFile.png", "rb") as fh:
    buf = io.BytesIO(fh.read())

upload_file = waifuvault.FileUpload(buf, "aCoolFile.png")
upload_res = waifuvault.upload_file(upload_file)
print(f"{upload_res.url}")

Get File Info

If you have a token from your upload. Then you can get file info. This results in the following info:

  • token
  • url
  • protected
  • retentionPeriod

Use the file_info function. This function takes the following options as parameters:

Option Type Description Required Extra info
token string The token of the upload true
formatted boolean If you want the retentionPeriod to be human-readable or an epoch false defaults to false
import waifuvault
upload_info = waifuvault.file_info(your_token,False)
print(upload_info.retentionPeriod)
print(upload_info.url)

Human-readable timestamp:

import waifuvault
upload_info = waifuvault.file_info(your_token,True)
print(upload_info.retentionPeriod)
print(upload_info.url)

Update File Info

If you have a token from your upload, then you can update the information for the file. You can change the password or remove it, you can set custom expiry time or remove it, and finally you can choose whether the filename is hidden.

Use the file_update function. This function takes the following options as parameters:

Option Type Description Required Extra info
token string The token of the upload true
password string The current password of the file false Set to empty string to remove password
previous_password string The previous password of the file, if changing password false
custom_expiry string Custom expiry in the same form as upload command false Set to empty string to remove custom expiry
hide_filename bool Sets whether the filename is visible in the URL or not false
import waifuvault
update_info = waifuvault.file_update(your_token,custom_expiry="2d")
print(upload_info.retentionPeriod)

Delete File

To delete a file, you must supply your token to the delete_file function.

This function takes the following options as parameters:

Option Type Description Required Extra info
token string The token of the file you wish to delete true

NOTE: delete_file will only ever either return true or throw an exception if the token is invalid

import waifuvault
del_file = waifuvault.delete_file(your_token)
print(del_file)

Get File

This lib also supports obtaining a file from the API as a Buffer by supplying either the token or the unique identifier of the file (epoch/filename).

Use the get_file function. This function takes the following options an object:

Option Type Description Required Extra info
token string The token of the file you want to download true only if filename is not set if filename is set, then this can not be used
url string The URL of the file true only if token is not set if token is set, then this can not be used
password string The password for the file true if file is encrypted Passed as a parameter on the function call

Important! The Unique identifier filename is the epoch/filename only if the file uploaded did not have a hidden filename, if it did, then it's just the epoch. For example: 1710111505084/08.png is the Unique identifier for a standard upload of a file called 08.png, if this was uploaded with hidden filename, then it would be 1710111505084.png

Obtain an encrypted file

import waifuvault
upload_enc_res = FileResponse(your_token,None,False,None)
file_enc_down = waifuvault.get_file(upload_enc_res,"your_password")
print(file_enc_down.__sizeof__())

Obtain a file from URL

import waifuvault
upload_enc_res = FileResponse(None,your_url,False,None)
file_enc_down = waifuvault.get_file(upload_enc_res,"your_password")
print(file_enc_down.__sizeof__())

Create Bucket

Buckets are virtual collections that are linked to your IP and a token. When you create a bucket, you will receive a bucket token that you can use in Get Bucket to get all the files in that bucket

NOTE: Only one bucket is allowed per client IP address, if you call it more than once, it will return the same bucket token

To create a bucket, use the create_bucket function. This function does not take any arguments.

import waifuvault
bucket = waifuvault.create_bucket()
print(bucket.token)

Delete Bucket

Deleting a bucket will delete the bucket and all the files it contains.

IMPORTANT: All contained files will be DELETED along with the Bucket!

To delete a bucket, you must call the delete_bucket function with the following options as parameters:

Option Type Description Required Extra info
token string The token of the bucket to delete true

NOTE: deleteBucket will only ever either return true or throw an exception if the token is invalid

import waifuvault
resp = waifuvault.delete_bucket("some-bucket-token")
print(resp)

Get Bucket

To get the list of files contained in a bucket, you use the get_bucket functions and supply the token. This function takes the following options as parameters:

Option Type Description Required Extra info
token string The token of the bucket true

This will respond with the bucket and all the files the bucket contains.

import waifuvault
bucket = waifuvault.get_bucket("some-bucket-token")
print(bucket.token)
print(bucket.files)  # Array of file objects

Get Restrictions

To get the list of restrictions applied to the server, you use the get_restrictions functions.

This will respond with an array of name, value entries describing the restrictions applied to the server.

NOTE: Restrictions are cached for 10 minutes

import waifuvault
restrictions = waifuvault.get_restrictions()

print(restrictions.Restrictions)  # Array of restriction objects

Clear Restrictions

To clear the cached restrictions in the SDK, you use the clear_restrictions function.

This will remove the cached restrictions and a fresh copy will be downloaded at the next upload.

import waifuvault
waifuvault.clear_restrictions()