mirror of
https://github.com/ynput/ayon-core.git
synced 2025-12-25 13:24:54 +01:00
141 lines
3.4 KiB
Python
141 lines
3.4 KiB
Python
import os
|
|
import json
|
|
import ftrack_api
|
|
import appdirs
|
|
import getpass
|
|
try:
|
|
from urllib.parse import urlparse
|
|
except ImportError:
|
|
from urlparse import urlparse
|
|
|
|
|
|
CONFIG_PATH = os.path.normpath(appdirs.user_data_dir("pype-app", "pype"))
|
|
CREDENTIALS_FILE_NAME = "ftrack_cred.json"
|
|
CREDENTIALS_PATH = os.path.join(CONFIG_PATH, CREDENTIALS_FILE_NAME)
|
|
CREDENTIALS_FOLDER = os.path.dirname(CREDENTIALS_PATH)
|
|
|
|
if not os.path.isdir(CREDENTIALS_FOLDER):
|
|
os.makedirs(CREDENTIALS_FOLDER)
|
|
|
|
USER_GETTER = None
|
|
|
|
|
|
def get_ftrack_hostname(ftrack_server=None):
|
|
if not ftrack_server:
|
|
ftrack_server = os.environ["FTRACK_SERVER"]
|
|
|
|
if "//" not in ftrack_server:
|
|
ftrack_server = "//" + ftrack_server
|
|
|
|
return urlparse(ftrack_server).hostname
|
|
|
|
|
|
def get_user():
|
|
if USER_GETTER:
|
|
return USER_GETTER()
|
|
return getpass.getuser()
|
|
|
|
|
|
def get_credentials(ftrack_server=None, user=None):
|
|
credentials = {}
|
|
if not os.path.exists(CREDENTIALS_PATH):
|
|
with open(CREDENTIALS_PATH, "w") as file:
|
|
file.write(json.dumps(credentials))
|
|
file.close()
|
|
return credentials
|
|
|
|
with open(CREDENTIALS_PATH, "r") as file:
|
|
content = file.read()
|
|
|
|
hostname = get_ftrack_hostname(ftrack_server)
|
|
if not user:
|
|
user = get_user()
|
|
|
|
content_json = json.loads(content or "{}")
|
|
credentials = content_json.get(hostname, {}).get(user) or {}
|
|
|
|
return credentials
|
|
|
|
|
|
def save_credentials(ft_user, ft_api_key, ftrack_server=None, user=None):
|
|
hostname = get_ftrack_hostname(ftrack_server)
|
|
if not user:
|
|
user = get_user()
|
|
|
|
with open(CREDENTIALS_PATH, "r") as file:
|
|
content = file.read()
|
|
|
|
content_json = json.loads(content or "{}")
|
|
if hostname not in content_json:
|
|
content_json[hostname] = {}
|
|
|
|
content_json[hostname][user] = {
|
|
"username": ft_user,
|
|
"api_key": ft_api_key
|
|
}
|
|
|
|
# Deprecated keys
|
|
if "username" in content_json:
|
|
content_json.pop("username")
|
|
if "apiKey" in content_json:
|
|
content_json.pop("apiKey")
|
|
|
|
with open(CREDENTIALS_PATH, "w") as file:
|
|
file.write(json.dumps(content_json, indent=4))
|
|
|
|
|
|
def clear_credentials(ft_user=None, ftrack_server=None, user=None):
|
|
if not ft_user:
|
|
ft_user = os.environ.get("FTRACK_API_USER")
|
|
|
|
if not ft_user:
|
|
return
|
|
|
|
hostname = get_ftrack_hostname(ftrack_server)
|
|
if not user:
|
|
user = get_user()
|
|
|
|
with open(CREDENTIALS_PATH, "r") as file:
|
|
content = file.read()
|
|
|
|
content_json = json.loads(content or "{}")
|
|
if hostname not in content_json:
|
|
content_json[hostname] = {}
|
|
|
|
content_json[hostname].pop(user, None)
|
|
|
|
with open(CREDENTIALS_PATH, "w") as file:
|
|
file.write(json.dumps(content_json))
|
|
|
|
|
|
def set_env(ft_user=None, ft_api_key=None):
|
|
os.environ["FTRACK_API_USER"] = ft_user or ""
|
|
os.environ["FTRACK_API_KEY"] = ft_api_key or ""
|
|
|
|
|
|
def get_env_credentials():
|
|
return (
|
|
os.environ.get("FTRACK_API_USER"),
|
|
os.environ.get("FTRACK_API_KEY")
|
|
)
|
|
|
|
|
|
def check_credentials(ft_user, ft_api_key, ftrack_server=None):
|
|
if not ftrack_server:
|
|
ftrack_server = os.environ["FTRACK_SERVER"]
|
|
|
|
if not ft_user or not ft_api_key:
|
|
return False
|
|
|
|
try:
|
|
session = ftrack_api.Session(
|
|
server_url=ftrack_server,
|
|
api_key=ft_api_key,
|
|
api_user=ft_user
|
|
)
|
|
session.close()
|
|
|
|
except Exception:
|
|
return False
|
|
|
|
return True
|