Блог

я использовал пакет python (spotify_dl) для создания некоторых файлов. я делал это вручную через командную строку: spotify_dl-l spotify_playlist_link-o каталог загрузки

теперь я хочу сделать это из другого скрипта python.

я просмотрел код пакета и нашел основную функцию, но я не могу понять, как запустить основную функцию с выбранными мной аргументами — на python.

например, чего я хочу:

 from spotify_dl import spotify_dl as sp_dl

if __name__=='__main__':
    destination_dir = r'D:somefolderpath'
    playlists_url = ['url1','url2','url3',....]

    for url in playlists_url:
        sp_dl.spotify_dl(l=url,o=destination_dir)
 

для большей ясности ниже приведен фактический код основной функции spotify_dl:

 #!/usr/bin/env python
import argparse
import json
import os
import sys
from logging import DEBUG
from pathlib import Path, PurePath

import spotipy
from spotipy.oauth2 import SpotifyClientCredentials

from spotify_dl.constants import VERSION
from spotify_dl.models import db, Song
from spotify_dl.scaffold import log, check_for_tokens
from spotify_dl.spotify import fetch_tracks, parse_spotify_url, validate_spotify_url, get_item_name
from spotify_dl.youtube import download_songs, default_filename, playlist_num_filename


def spotify_dl():
    """Main entry point of the script."""
    parser = argparse.ArgumentParser(prog='spotify_dl')
    parser.add_argument('-l', '--url', action="store",
                        help="Spotify Playlist link URL", type=str, required=True)
    parser.add_argument('-o', '--output', type=str, action='store',
                        help='Specify download directory.', required=True)
    parser.add_argument('-d', '--download', action='store_true',
                        help='Download using youtube-dl', default=True)
    parser.add_argument('-f', '--format_str', type=str, action='store',
                        help='Specify youtube-dl format string.',
                        default='bestaudio/best')
    parser.add_argument('-k', '--keep_playlist_order', default=False,
                        action='store_true',
                        help='Whether to keep original playlist ordering or not.')
    parser.add_argument('-m', '--skip_mp3', action='store_true',
                        help='Don't convert downloaded songs to mp3')
    parser.add_argument('-s', '--scrape', action="store",
                        help="Use HTML Scraper for YouTube Search", default=True)
    parser.add_argument('-V', '--verbose', action='store_true',
                        help='Show more information on what''s happening.')
    parser.add_argument('-v', '--version', action='store_true',
                        help='Shows current version of the program')
    args = parser.parse_args()

    if args.version:
        print("spotify_dl v{}".format(VERSION))
        exit(0)

    db.connect()
    db.create_tables([Song])
    if os.path.isfile(os.path.expanduser('~/.spotify_dl_settings')):
        with open(os.path.expanduser('~/.spotify_dl_settings')) as file:
            config = json.loads(file.read())

        for key, value in config.items():
            if value and (value.lower() == 'true' or value.lower() == 't'):
                setattr(args, key, True)
            else:
                setattr(args, key, value)

    if args.verbose:
        log.setLevel(DEBUG)

    log.info('Starting spotify_dl')
    log.debug('Setting debug mode on spotify_dl')

    if not check_for_tokens():
        exit(1)

    sp = spotipy.Spotify(auth_manager=SpotifyClientCredentials())
    log.debug('Arguments: {}'.format(args))

    if args.url:
        valid_item = validate_spotify_url(args.url)

    if not valid_item:
        sys.exit(1)

    if args.output:
        item_type, item_id = parse_spotify_url(args.url)
        directory_name = get_item_name(sp, item_type, item_id)
        save_path = Path(PurePath.joinpath(Path(args.output), Path(directory_name)))
        save_path.mkdir(parents=True, exist_ok=True)
        log.info("Saving songs to: {}".format(directory_name))

    songs = fetch_tracks(sp, item_type, args.url)
    if args.download is True:
        file_name_f = default_filename
        if args.keep_playlist_order:
            file_name_f = playlist_num_filename

        download_songs(songs, save_path, args.format_str, args.skip_mp3, args.keep_playlist_order, file_name_f)


if __name__ == '__main__':
    spotify_dl()
 

до сих пор я видел упоминания о sys.argv, но также некоторые комментарии высказывались против его использования.
что мне нужно, так это четкий метод, который я мог бы запускать в цикле без осложнений.

spotify_dl плохо спроектирован, я обычно использую это:

 def parse_args(argv=None):
    parser = argparse.ArgumentParser()
    ...
    return parser.parse_args(argv)

def main(args):
    # run your code

if __name__ == "__main__":
    args = parse_args()
    main(args)
 

Затем вы можете импортировать этот первый сценарий в другой и вызвать main функцию:

 from my_first_script import main

def a_function():
    args = namedtuple("args", ("arg1", ...))("value1")
    main(args)
 

Но вы можете обойти это, переопределив sys.argv :

 if __name__=='__main__':
    destination_dir = r'D:somefolderpath'
    playlists_url = ['url1','url2','url3',....]

    for url in playlists_url:
        sys.argv = [sys.executable, "-o", destination_dir, "-l", url]
        sp_dl.spotify_dl()
 

Вы можете использовать subprocess здесь. Я не пробовал ваш пример, потому что у меня не установлены эти библиотеки, но вот простой пример.

Во-первых, простой скрипт командной строки, называемый testpy.py , имеет следующее:

 import sys
import argparse
def check_arg(args=None):
    parser = argparse.ArgumentParser(prog='Test',
                     description='Test',
                     epilog='Test')
    parser.add_argument('-f', '--first',
                        help='First argument',
                        required='True')
    results = parser.parse_args(args)
    return (results.first)



def main():
    with open('test.txt', 'a') as file:
        file.write('Success, {}n'.format(f))


if __name__ == '__main__':
    f = check_arg(sys.argv[1:])
    main()
 

Это принимает один аргумент f и экспортирует текстовый файл (test.txt) с любым аргументом, который был представлен.

Этот сценарий можно запустить из следующего сценария как (сохраните его как новый файл сценария).:

 from subprocess import run, PIPE
args = ['python3', 'testpy.py', '-f', 'First input argument text']
res = run(args, stdout=PIPE, stderr=PIPE)
 

Если вы запустите это, вы увидите, что файл test.txt будет экспортирован и вызов сценария testpy.py из этого сценария прошел успешно.