Make code more pythonic (#75)
* Make code more pythonic * Make code more pythonic --------- Co-authored-by: Nicola Canepa <canne74@gmail.com>
This commit is contained in:
Nicola
207
main.py
207
main.py
@@ -16,7 +16,8 @@ def take_boolean_input(default=True):
|
||||
return False
|
||||
print('Please answer with y or n.', end=' ')
|
||||
|
||||
def take_input(service_name, service_type):
|
||||
def take_input(service_name, service_type, name_to_show=None):
|
||||
print(f'Use {name_to_show or service_name.capitalize()}? [Y/n]', end=" ")
|
||||
choice = take_boolean_input()
|
||||
if choice:
|
||||
services_classed[service_type].append(service_name)
|
||||
@@ -33,7 +34,7 @@ def take_directory_input():
|
||||
print('Please make sure the path is absolute, meaning it starts at the root of your filesystem and starts with "/":', end=' ')
|
||||
|
||||
def get_system_timezone():
|
||||
tz_path = "/etc/localtime"
|
||||
tz_path = '/etc/localtime'
|
||||
|
||||
if os.path.exists(tz_path):
|
||||
if os.path.islink(tz_path):
|
||||
@@ -41,130 +42,116 @@ def get_system_timezone():
|
||||
return tz.split('zoneinfo/')[-1]
|
||||
return None
|
||||
|
||||
print('Welcome to the EZarr CLI.')
|
||||
print('This CLI will ask you which services you\'d like to use and more. If you\'d like more information about a '
|
||||
'certain service, look in the README.')
|
||||
def main():
|
||||
print('Welcome to the EZarr CLI.')
|
||||
print("This CLI will ask you which services you'd like to use and more. If you'd like more information about a "
|
||||
'certain service, look in the README.')
|
||||
|
||||
print('\n===SERVARR===')
|
||||
services_classed['servarr'] = []
|
||||
print('Use Sonarr? [Y/n]', end=" ")
|
||||
take_input('sonarr', 'servarr')
|
||||
print('Use Radarr? [Y/n]', end=" ")
|
||||
take_input('radarr', 'servarr')
|
||||
print('Use Lidarr? [Y/n]', end=" ")
|
||||
take_input('lidarr', 'servarr')
|
||||
print('Use Readarr? [Y/n]', end=" ")
|
||||
take_input('readarr', 'servarr')
|
||||
print('Use Mylar3? [Y/n]', end=" ")
|
||||
take_input('mylar3', 'servarr')
|
||||
print('Use Audiobookshelf? [Y/n]', end=" ")
|
||||
take_input('audiobookshelf', 'servarr')
|
||||
print('Use Homarr? [Y/n]', end=" ")
|
||||
take_input('homarr', 'servarr')
|
||||
if len(services_classed['servarr']) == 0:
|
||||
print('Warning: no media management services selected.')
|
||||
if services_classed['servarr'].__contains__('sonarr') or services_classed['servarr'].__contains__('radarr'):
|
||||
print('Use Bazarr? [Y/n]', end=" ")
|
||||
take_input('bazarr', 'servarr')
|
||||
print('\n===SERVARR===')
|
||||
services_classed['servarr'] = []
|
||||
take_input('sonarr', 'servarr')
|
||||
take_input('radarr', 'servarr')
|
||||
take_input('lidarr', 'servarr')
|
||||
take_input('readarr', 'servarr')
|
||||
take_input('mylar3', 'servarr')
|
||||
take_input('audiobookshelf', 'servarr')
|
||||
take_input('homarr', 'servarr')
|
||||
if len(services_classed['servarr']) == 0:
|
||||
print('Warning: no media management services selected.')
|
||||
if 'sonarr' in services_classed['servarr'] or 'radarr' in services_classed['servarr']:
|
||||
take_input('bazarr', 'servarr')
|
||||
|
||||
print('\n===INDEXERS===')
|
||||
services_classed['indexer'] = []
|
||||
print('Use Prowlarr? [Y/n]', end=" ")
|
||||
take_input('prowlarr', 'indexer')
|
||||
print('Use Jackett? [Y/n]', end=" ")
|
||||
take_input('jackett', 'indexer')
|
||||
if len(services_classed['indexer']) == 0:
|
||||
print('Warning: no indexing service selected.')
|
||||
print('\n===INDEXERS===')
|
||||
services_classed['indexer'] = []
|
||||
take_input('prowlarr', 'indexer')
|
||||
take_input('jackett', 'indexer')
|
||||
if len(services_classed['indexer']) == 0:
|
||||
print('Warning: no indexing service selected.')
|
||||
|
||||
print('\n===CLOUDFLARE BYPASS===')
|
||||
services_classed['bypass'] = []
|
||||
print('Use Flaresolverr? [Y/n]', end=" ")
|
||||
take_input('flaresolverr', 'bypass')
|
||||
print('\n===CLOUDFLARE BYPASS===')
|
||||
services_classed['bypass'] = []
|
||||
take_input('flaresolverr', 'bypass')
|
||||
|
||||
print('\n===MEDIA SERVERS===')
|
||||
services_classed['ms'] = []
|
||||
print('Use PleX? [Y/n]', end=" ")
|
||||
take_input('plex', 'ms')
|
||||
if services_classed['ms'].__contains__('plex'):
|
||||
print('Use Tautulli? [Y/n]', end=" ")
|
||||
take_input('tautulli', 'ms')
|
||||
if services_classed['servarr'].__contains__('sonarr') or services_classed['servarr'].__contains__('radarr'):
|
||||
print('Use Overseerr? [Y/n]', end=" ")
|
||||
take_input('overseerr', 'servarr')
|
||||
print('Use Jellyfin? [Y/n]', end=" ")
|
||||
take_input('jellyfin', 'ms')
|
||||
if (services_classed['ms'].__contains__('jellyfin')
|
||||
and (services_classed['servarr'].__contains__('sonarr') or services_classed['servarr'].__contains__('radarr'))):
|
||||
print('Use Jellyseerr? [Y/n]', end=" ")
|
||||
take_input('jellyseerr', 'servarr')
|
||||
if len(services_classed['ms']) == 0:
|
||||
print('Warning: no media servers selected.')
|
||||
print('\n===MEDIA SERVERS===')
|
||||
services_classed['ms'] = []
|
||||
take_input('plex', 'ms')
|
||||
if 'plex' in services_classed['ms']:
|
||||
take_input('tautulli', 'ms')
|
||||
if 'sonarr' in services_classed['servarr'] or 'radarr' in services_classed['servarr']:
|
||||
take_input('overseerr', 'servarr')
|
||||
take_input('jellyfin', 'ms')
|
||||
if ('jellyfin' in services_classed['ms']
|
||||
and ('sonarr' in services_classed['servarr'] or 'radarr' in services_classed['servarr'])):
|
||||
take_input('jellyseerr', 'servarr')
|
||||
if len(services_classed['ms']) == 0:
|
||||
print('Warning: no media servers selected.')
|
||||
|
||||
print('\n===BITTORRENT===')
|
||||
services_classed['torrent'] = []
|
||||
print('Use qBittorrent? [Y/n]', end=" ")
|
||||
take_input('qbittorrent', 'torrent')
|
||||
print('\n===BITTORRENT===')
|
||||
services_classed['torrent'] = []
|
||||
take_input('qbittorrent', 'torrent', 'qBittorrent')
|
||||
|
||||
print('\n===USENET===')
|
||||
services_classed['usenet'] = []
|
||||
print('Use SABnzbd? [Y/n]', end=" ")
|
||||
take_input('sabnzbd', 'usenet')
|
||||
print('\n===USENET===')
|
||||
services_classed['usenet'] = []
|
||||
take_input('sabnzbd', 'usenet', 'SABnzbd')
|
||||
|
||||
if len(services_classed['torrent']) == 0 and len(services_classed['usenet']) == 0:
|
||||
print('Warning: no usenet or BitTorrent clients selected.')
|
||||
if len(services_classed['torrent']) == 0 and len(services_classed['usenet']) == 0:
|
||||
print('Warning: no usenet or BitTorrent clients selected.')
|
||||
|
||||
services = []
|
||||
for service_class in services_classed.keys():
|
||||
services.extend(services_classed[service_class])
|
||||
if len(services) == 0:
|
||||
print('No services selected. Terminating.')
|
||||
exit(1)
|
||||
services = []
|
||||
for service_class in services_classed.keys():
|
||||
services.extend(services_classed[service_class])
|
||||
if len(services) == 0:
|
||||
print('No services selected. Terminating.')
|
||||
exit(1)
|
||||
|
||||
print('\n===CONFIGURATION===')
|
||||
print('\n===CONFIGURATION===')
|
||||
|
||||
print('Please enter your timezone (like "Europe/Amsterdam") or press enter to use your system\'s configured timezone:', end=' ')
|
||||
timezone = input()
|
||||
if timezone == '':
|
||||
timezone = get_system_timezone()
|
||||
print('Please enter your timezone (like "Europe/Amsterdam") or press enter to use your system\'s configured timezone:', end=' ')
|
||||
timezone = input()
|
||||
if timezone == '':
|
||||
timezone = get_system_timezone()
|
||||
|
||||
if len(timezone) == 0: # if user pressed enter and reading timezone from /etc/localtime failed then default to Amsterdam
|
||||
timezone = 'Europe/Amsterdam'
|
||||
if len(timezone) == 0: # if user pressed enter and reading timezone from /etc/localtime failed then default to Amsterdam
|
||||
timezone = 'Europe/Amsterdam'
|
||||
|
||||
plex_claim = ''
|
||||
if services.__contains__('plex'):
|
||||
print('If you have a PleX claim token, enter it now. Otherwise, just press enter.', end=' ')
|
||||
plex_claim = input()
|
||||
plex_claim = ''
|
||||
if 'plex' in services:
|
||||
print('If you have a PleX claim token, enter it now. Otherwise, just press enter.', end=' ')
|
||||
plex_claim = input()
|
||||
|
||||
print('Where would you like to keep your files?', end=' ')
|
||||
root_dir = take_directory_input()
|
||||
print('Where would you like to keep your files?', end=' ')
|
||||
root_dir = take_directory_input()
|
||||
|
||||
compose = open('docker-compose.yml', 'w')
|
||||
compose.write(
|
||||
'---\n'
|
||||
'services:\n'
|
||||
)
|
||||
compose = open('docker-compose.yml', 'w')
|
||||
compose.write(
|
||||
'---\n'
|
||||
'services:\n'
|
||||
)
|
||||
|
||||
container_config = ContainerConfig(root_dir, timezone, plex_claim=plex_claim)
|
||||
container_config = ContainerConfig(root_dir, timezone, plex_claim=plex_claim)
|
||||
|
||||
for service in services:
|
||||
compose.write(getattr(container_config, service)())
|
||||
compose.close()
|
||||
print("Docker compose file generated successfully.")
|
||||
|
||||
print("Do you want to also generate the required folder structure and permissions? (this is required for first time setup) [Y/n]: ")
|
||||
generate_permissions = take_boolean_input()
|
||||
if generate_permissions:
|
||||
permission_setup = UserGroupSetup(root_dir=root_dir)
|
||||
for service in services:
|
||||
try:
|
||||
getattr(permission_setup, service)()
|
||||
except AttributeError:
|
||||
pass
|
||||
else:
|
||||
print("Permission and folder structure generation skipped by user.")
|
||||
compose.write(getattr(container_config, service)())
|
||||
compose.close()
|
||||
print("Docker compose file generated successfully.")
|
||||
|
||||
print("Do you want to also generate the required folder structure and permissions? (this is required for first time setup) [Y/n]: ")
|
||||
generate_permissions = take_boolean_input()
|
||||
if generate_permissions:
|
||||
permission_setup = UserGroupSetup(root_dir=root_dir)
|
||||
for service in services:
|
||||
try:
|
||||
getattr(permission_setup, service)()
|
||||
except AttributeError:
|
||||
pass
|
||||
else:
|
||||
print("Permission and folder structure generation skipped by user.")
|
||||
|
||||
|
||||
print('Process complete. You can now run "docker compose up -d" to start your containers.')
|
||||
print('Thank you for using EZarr. If you experience any issues or have feature requests, add them to our issues.')
|
||||
print('For questions, you can also use the discussions tab.')
|
||||
exit(0)
|
||||
print('Process complete. You can now run "docker compose up -d" to start your containers.')
|
||||
print('Thank you for using EZarr. If you experience any issues or have feature requests, add them to our issues.')
|
||||
print('For questions, you can also use the discussions tab.')
|
||||
exit(0)
|
||||
|
||||
if __name__ == '__main__':
|
||||
main()
|
||||
|
||||
Reference in New Issue
Block a user