Adjusted code from previous PR

This commit is contained in:
RemixDev 2020-05-20 17:02:24 +02:00
parent 09ca778e51
commit 012db6f572
2 changed files with 12 additions and 9 deletions

View File

@ -8,10 +8,10 @@ from os.path import isfile
@click.command() @click.command()
@click.option('-b', '--bitrate', default=None, help='Overwrites the default bitrate selected') @click.option('-b', '--bitrate', default=None, help='Overwrites the default bitrate selected')
@click.option('-bot', '--botmode', is_flag=True, help='Enables bot mode') @click.option('-l', '--local', is_flag=True, help='Downloads in a local folder insted of using the default')
@click.argument('url', nargs=-1, required=True) @click.argument('url', nargs=-1, required=True)
def download(bitrate, botmode, url): def download(bitrate, localFolder, url):
settings = initSettings(botmode) settings = initSettings(localFolder)
app.login() app.login()
if isfile(url[0]): if isfile(url[0]):
filename = url[0] filename = url[0]
@ -20,7 +20,7 @@ def download(bitrate, botmode, url):
for u in url: for u in url:
app.downloadLink(u, settings, bitrate) app.downloadLink(u, settings, bitrate)
click.echo("All done!") click.echo("All done!")
if botmode: if localFolder:
click.echo(settings['downloadLocation']) #folder name output click.echo(settings['downloadLocation']) #folder name output

View File

@ -4,6 +4,10 @@ import os.path as path
from os import makedirs from os import makedirs
import random import random
import string import string
import logging
logging.basicConfig(level=logging.INFO)
logger = logging.getLogger('deemix')
import deemix.utils.localpaths as localpaths import deemix.utils.localpaths as localpaths
@ -11,7 +15,7 @@ settings = {}
defaultSettings = {} defaultSettings = {}
def initSettings(bot_mode = False): def initSettings(localFolder = False):
global settings global settings
global defaultSettings global defaultSettings
currentFolder = path.abspath(path.dirname(__file__)) currentFolder = path.abspath(path.dirname(__file__))
@ -27,12 +31,11 @@ def initSettings(bot_mode = False):
settings = json.load(configFile) settings = json.load(configFile)
settingsCheck() settingsCheck()
if bot_mode: if localFolder:
print("Im using bot mode")
settings['downloadLocation'] = randomString(12) settings['downloadLocation'] = randomString(12)
logger.info("Using a local download folder: "+settings['downloadLocation'])
elif settings['downloadLocation'] == "": elif settings['downloadLocation'] == "":
print("I'm using normal mode") settings['downloadLocation'] = path.join(localpaths.getHomeFolder(), 'deemix Music')
settings['downloadLocation'] = path.join(localpaths.getHomeFolder(), 'deemix Music')
saveSettings(settings) saveSettings(settings)
makedirs(settings['downloadLocation'], exist_ok=True) makedirs(settings['downloadLocation'], exist_ok=True)
return settings return settings