feat: Add nextcloud and immich api integration
This commit is contained in:
49
tools/cloud.py
Normal file
49
tools/cloud.py
Normal file
@@ -0,0 +1,49 @@
|
||||
import os
|
||||
import requests
|
||||
|
||||
login = os.getenv("WOODBURN_USER")
|
||||
|
||||
|
||||
def getUserQuota(user: str) -> dict[str, int | str]:
|
||||
"""
|
||||
Get the user's quota information from the environment variable.
|
||||
Returns a dictionary with 'used' and 'total' as integers.
|
||||
"""
|
||||
headers = {"OCS-APIRequest": "true", "Accept": "application/json"}
|
||||
if not login:
|
||||
return {
|
||||
"used": 0,
|
||||
"total": 5,
|
||||
"percentage": "0.0",
|
||||
"error": "WOODBURN_USER environment variable not set",
|
||||
}
|
||||
# curl -u user https://cloud.woodburn.au/ocs/v1.php/cloud/users/nathan OCS-APIRequest:true
|
||||
response = requests.get(
|
||||
f"https://cloud.woodburn.au/ocs/v1.php/cloud/users/{user}",
|
||||
headers=headers,
|
||||
auth=(login.split(":")[0], login.split(":")[1]),
|
||||
)
|
||||
if response.status_code != 200:
|
||||
return {
|
||||
"used": 0,
|
||||
"total": 5,
|
||||
"percentage": "0.0",
|
||||
"error": f"Failed to fetch quota: {response.status_code}",
|
||||
}
|
||||
data = response.json()
|
||||
# If the request failed
|
||||
if data.get("ocs", {}).get("meta", {}).get("status") != "ok":
|
||||
return {
|
||||
"used": 0,
|
||||
"total": 5,
|
||||
"percentage": "0.0",
|
||||
"error": data.get("ocs", {})
|
||||
.get("meta", {})
|
||||
.get("message", "Unknown error"),
|
||||
}
|
||||
|
||||
quota = data.get("ocs", {}).get("data", {}).get("quota", {})
|
||||
# Convert to GB
|
||||
used = int(quota.get("used", 0)) // (1024 * 1024 * 1024)
|
||||
total = int(quota.get("total", 0)) // (1024 * 1024 * 1024)
|
||||
return {"used": used, "total": total, "percentage": quota.get("relative", "0.0")}
|
||||
Reference in New Issue
Block a user