3
3
from dotenv import load_dotenv , find_dotenv
4
4
from datetime import datetime , timedelta
5
5
import logging
6
- from logging .handlers import RotatingFileHandler , TimedRotatingFileHandler , BaseRotatingHandler
6
+ from logging .handlers import (
7
+ RotatingFileHandler ,
8
+ TimedRotatingFileHandler ,
9
+ BaseRotatingHandler ,
10
+ )
7
11
from packaging .version import Version , parse as parse_version
8
12
import time
9
13
import os
31
35
from ruamel .yaml import YAML
32
36
33
37
34
- load_dotenv (find_dotenv ('./config/.env' ))
35
-
38
+ load_dotenv (find_dotenv ("./config/.env" ))
39
+
40
+
36
41
def load_secret_or_env (secret_name , default = None ):
37
- secret_file = f' /run/secrets/{ secret_name } '
42
+ secret_file = f" /run/secrets/{ secret_name } "
38
43
try :
39
- with open (secret_file , 'r' ) as file :
44
+ with open (secret_file , "r" ) as file :
40
45
return file .read ().strip ()
41
46
except IOError :
42
47
return os .getenv (secret_name .upper (), default )
43
48
49
+
44
50
def get_valid_env (env_var , default ):
45
51
value = os .getenv (env_var )
46
52
if value is None or value .strip () == "" :
@@ -51,65 +57,66 @@ def get_valid_env(env_var, default):
51
57
elif isinstance (default , float ):
52
58
return float (value )
53
59
elif isinstance (default , bool ):
54
- return value .lower () in [' true' , '1' , ' yes' ]
60
+ return value .lower () in [" true" , "1" , " yes" ]
55
61
else :
56
62
return value
57
63
except ValueError :
58
64
return default
59
65
60
- PLEXTOKEN = load_secret_or_env ('plex_token' )
61
- PLEXADD = load_secret_or_env ('plex_address' )
62
- RDAPIKEY = load_secret_or_env ('rd_api_key' )
63
- ADAPIKEY = load_secret_or_env ('ad_api_key' )
64
- ZURGUSER = load_secret_or_env ('zurg_user' )
65
- ZURGPASS = load_secret_or_env ('zurg_pass' )
66
- GHTOKEN = load_secret_or_env ('github_token' )
67
- SEERRAPIKEY = load_secret_or_env ('seerr_api_key' )
68
- SEERRADD = load_secret_or_env ('seerr_address' )
69
- PGADMINEMAIL = load_secret_or_env ('pgadmin_setup_email' )
70
- PGADMINPASS = load_secret_or_env ('pgadmin_setup_password' )
71
- DUPECLEAN = os .getenv ('DUPLICATE_CLEANUP' )
72
- CLEANUPINT = os .getenv ('CLEANUP_INTERVAL' )
66
+
67
+ PLEXTOKEN = load_secret_or_env ("plex_token" )
68
+ PLEXADD = load_secret_or_env ("plex_address" )
69
+ RDAPIKEY = load_secret_or_env ("rd_api_key" )
70
+ ADAPIKEY = load_secret_or_env ("ad_api_key" )
71
+ ZURGUSER = load_secret_or_env ("zurg_user" )
72
+ ZURGPASS = load_secret_or_env ("zurg_pass" )
73
+ GHTOKEN = load_secret_or_env ("github_token" )
74
+ SEERRAPIKEY = load_secret_or_env ("seerr_api_key" )
75
+ SEERRADD = load_secret_or_env ("seerr_address" )
76
+ PGADMINEMAIL = load_secret_or_env ("pgadmin_setup_email" )
77
+ PGADMINPASS = load_secret_or_env ("pgadmin_setup_password" )
78
+ DUPECLEAN = os .getenv ("DUPLICATE_CLEANUP" )
79
+ CLEANUPINT = os .getenv ("CLEANUP_INTERVAL" )
73
80
RCLONEMN = os .getenv ("RCLONE_MOUNT_NAME" )
74
81
RCLONELOGLEVEL = os .getenv ("RCLONE_LOG_LEVEL" )
75
82
ZURG = os .getenv ("ZURG_ENABLED" )
76
83
ZURGVERSION = os .getenv ("ZURG_VERSION" )
77
84
ZURGLOGLEVEL = os .getenv ("ZURG_LOG_LEVEL" )
78
- ZURGUPDATE = os .getenv (' ZURG_UPDATE' )
79
- NFSMOUNT = os .getenv (' NFS_ENABLED' )
80
- NFSPORT = os .getenv (' NFS_PORT' )
81
- ZURGPORT = os .getenv (' ZURG_PORT' )
85
+ ZURGUPDATE = os .getenv (" ZURG_UPDATE" )
86
+ NFSMOUNT = os .getenv (" NFS_ENABLED" )
87
+ NFSPORT = os .getenv (" NFS_PORT" )
88
+ ZURGPORT = os .getenv (" ZURG_PORT" )
82
89
RIVENBACKEND = os .getenv ("RIVEN_BACKEND_ENABLED" )
83
90
RIVENFRONTEND = os .getenv ("RIVEN_FRONTEND_ENABLED" )
84
91
RIVEN = os .getenv ("RIVEN_ENABLED" )
85
- RUPDATE = os .getenv (' RIVEN_UPDATE' )
86
- RIVENLOGLEVEL = os .getenv (' RIVEN_LOG_LEVEL' )
87
- FRONTENDLOGS = os .getenv (' FRONTEND_LOGS' )
88
- BACKENDLOGS = os .getenv (' BACKEND_LOGS' )
89
- RIVENBACKENDURL = os .getenv (' RIVEN_BACKEND_URL' )
90
- RIVENDATABASEHOST = os .getenv (' RIVEN_DATABASE_HOST' )
91
- RIVENDATABASEURL = os .getenv (' RIVEN_DATABASE_URL' )
92
- RBBRANCH = os .getenv (' RIVEN_BACKEND_BRANCH' )
93
- RFBRANCH = os .getenv (' RIVEN_FRONTEND_BRANCH' )
94
- RFOWNER = os .getenv (' RIVEN_FRONTEND_OWNER' )
95
- RBVERSION = os .getenv (' RIVEN_BACKEND_VERSION' )
96
- RFVERSION = os .getenv (' RIVEN_FRONTEND_VERSION' )
97
- RBUPDATE = os .getenv (' RIVEN_BACKEND_UPDATE' )
98
- RFUPDATE = os .getenv (' RIVEN_FRONTEND_UPDATE' )
99
- RFDIALECT = os .getenv (' RIVEN_FRONTEND_DIALECT' )
100
- SYMLINKLIBRARYPATH = os .getenv (' SYMLINK_LIBRARY_PATH' )
101
- SYMLINKRCLONEPATH = os .getenv (' SYMLINK_RCLONE_PATH' )
102
- RCLONELOGS = os .getenv (' RCLONE_LOGS' )
103
- db_host = ' 127.0.0.1'
104
- user_id = get_valid_env (' PUID' , 1001 )
105
- group_id = get_valid_env (' PGID' , 1001 )
106
- postgres_data = get_valid_env (' POSTGRES_DATA' , ' /postgres_data' )
107
- postgres_user = get_valid_env (' POSTGRES_USER' , ' DMB' )
108
- postgres_password = get_valid_env (' POSTGRES_PASSWORD' , ' postgres' )
109
- postgres_db = get_valid_env (' POSTGRES_DB' , ' riven' )
110
- RCLONEDIR = get_valid_env (' RCLONE_DIR' , ' /data' )
111
- ZILEAN = os .getenv (' ZILEAN_ENABLED' )
112
- ZILEANBRANCH = os .getenv (' ZILEAN_BRANCH' )
113
- ZILEANVERSION = get_valid_env (' ZILEAN_VERSION' , None )
114
- ZILEANUPDATE = os .getenv (' ZILEAN_UPDATE' )
115
- ZILEANLOGS = os .getenv (' ZILEAN_LOGS' )
92
+ RUPDATE = os .getenv (" RIVEN_UPDATE" )
93
+ RIVENLOGLEVEL = os .getenv (" RIVEN_LOG_LEVEL" )
94
+ FRONTENDLOGS = os .getenv (" FRONTEND_LOGS" )
95
+ BACKENDLOGS = os .getenv (" BACKEND_LOGS" )
96
+ RIVENBACKENDURL = os .getenv (" RIVEN_BACKEND_URL" )
97
+ RIVENDATABASEHOST = os .getenv (" RIVEN_DATABASE_HOST" )
98
+ RIVENDATABASEURL = os .getenv (" RIVEN_DATABASE_URL" )
99
+ RBBRANCH = os .getenv (" RIVEN_BACKEND_BRANCH" )
100
+ RFBRANCH = os .getenv (" RIVEN_FRONTEND_BRANCH" )
101
+ RFOWNER = os .getenv (" RIVEN_FRONTEND_OWNER" )
102
+ RBVERSION = os .getenv (" RIVEN_BACKEND_VERSION" )
103
+ RFVERSION = os .getenv (" RIVEN_FRONTEND_VERSION" )
104
+ RBUPDATE = os .getenv (" RIVEN_BACKEND_UPDATE" )
105
+ RFUPDATE = os .getenv (" RIVEN_FRONTEND_UPDATE" )
106
+ RFDIALECT = os .getenv (" RIVEN_FRONTEND_DIALECT" )
107
+ SYMLINKLIBRARYPATH = os .getenv (" SYMLINK_LIBRARY_PATH" )
108
+ SYMLINKRCLONEPATH = os .getenv (" SYMLINK_RCLONE_PATH" )
109
+ RCLONELOGS = os .getenv (" RCLONE_LOGS" )
110
+ db_host = " 127.0.0.1"
111
+ user_id = get_valid_env (" PUID" , 1001 )
112
+ group_id = get_valid_env (" PGID" , 1001 )
113
+ postgres_data = get_valid_env (" POSTGRES_DATA" , " /postgres_data" )
114
+ postgres_user = get_valid_env (" POSTGRES_USER" , " DMB" )
115
+ postgres_password = get_valid_env (" POSTGRES_PASSWORD" , " postgres" )
116
+ postgres_db = get_valid_env (" POSTGRES_DB" , " riven" )
117
+ RCLONEDIR = get_valid_env (" RCLONE_DIR" , " /data" )
118
+ ZILEAN = os .getenv (" ZILEAN_ENABLED" )
119
+ ZILEANBRANCH = os .getenv (" ZILEAN_BRANCH" )
120
+ ZILEANVERSION = get_valid_env (" ZILEAN_VERSION" , None )
121
+ ZILEANUPDATE = os .getenv (" ZILEAN_UPDATE" )
122
+ ZILEANLOGS = os .getenv (" ZILEAN_LOGS" )
0 commit comments