Das deutsche Python-Forum

Code: Alles auswählen

#!/usr/bin/env python

heavyRotationPercentage = 1.0 / 2.0

import os
import random
import pickle
import datetime

directory = os.path.dirname(os.path.abspath(__file__))
heavyRotationDir = os.path.join(directory, "HeavyRotation")
normalRotationDir = os.path.join(directory, "NormalRotation")
oldRotationDir = os.path.join(directory, "OldRotation")
jinglesDir = os.path.join(directory, "Jingles")
statusFile = os.path.join(directory, "status.txt")
currentHour = datetime.datetime.now().hour

status = [('','')]*8

if os.path.isfile(statusFile):
    with open(statusFile, "rb") as f:
        status = pickle.load(f)
lastFiles = [i[0] for i in status]

fileListFilter = lambda fileList: [fileName for fileName in fileList if
                                   fileName not in lastFiles and fileName[-4:] == ".mp3"]

choosenFileDir = False
choosenFileList = False


if status[-1][1]!=jinglesDir:
    #Letzte Track war kein Jingle
    choosenFileDir = jinglesDir
    choosenFileList = fileListFilter(os.listdir(jinglesDir))


if currentHour >= 10 and choosenFileDir == False:
    #Von 10:00 bis 24:00 aus der normalen / Heavy Rotation waehlen
    filesInHeavyRotation = fileListFilter(os.listdir(heavyRotationDir))
    filesInNormalRotation = fileListFilter(os.listdir(normalRotationDir))
    if random.random() <= heavyRotationPercentage and len(filesInHeavyRotation) > 0:
        choosenFileDir = heavyRotationDir
        choosenFileList = filesInHeavyRotation
    elif len(filesInNormalRotation) > 0:
        choosenFileDir = normalRotationDir
        choosenFileList = filesInNormalRotation

if choosenFileDir == False:
    # Von 00:00 bis 10:00 (oder wenn nichts anderes gefunden wurde) aus der alten Rotation waehlen
    choosenFileDir = oldRotationDir
    choosenFileList = fileListFilter(os.listdir(oldRotationDir))


choosenFile = random.choice(choosenFileList)

with open(statusFile, "wb") as f:
    status.append((choosenFile, choosenFileDir))
    pickle.dump(status[1:],f)

print os.path.join(choosenFileDir, choosenFile)

Code: Alles auswählen

#!/usr/bin/env python3
import datetime
import json
import random
from pathlib import Path

MAX_RECENTLY_PLAYED = 8
HEAVY_ROTATION_PERCENTAGE = 0.5

BASE_PATH = Path(__file__).absolute().parent
HEAVY_ROTATION_PATH = BASE_PATH / "HeavyRotation"
NORMAL_ROTATION_PATH = BASE_PATH / "NormalRotation"
OLD_ROTATION_PATH = BASE_PATH / "OldRotation"
JINGLES_PATH = BASE_PATH / "Jingles"
STATUS_FILE_PATH = BASE_PATH / "status.json"


def is_jingle(file_path):
    return file_path.parent == JINGLES_PATH


def get_fresh_file_paths(recently_played_file_paths, folder_path):
    return [
        path
        for path in folder_path.glob("*.mp3")
        if path not in recently_played_file_paths
    ]


def choose_rotation(recently_played_file_paths):
    if recently_played_file_paths and not is_jingle(
        recently_played_file_paths[-1]
    ):
        return get_fresh_file_paths(recently_played_file_paths, JINGLES_PATH)

    #
    # Von 10:00 bis 24:00 aus der normalen / Heavy Rotation wählen.
    #
    if datetime.datetime.now().hour >= 10:
        heavy_rotation_filenames = get_fresh_file_paths(
            recently_played_file_paths, HEAVY_ROTATION_PATH
        )
        if (
            heavy_rotation_filenames
            and random.random() <= HEAVY_ROTATION_PERCENTAGE
        ):
            return heavy_rotation_filenames

        normal_rotation_filenames = get_fresh_file_paths(
            recently_played_file_paths, NORMAL_ROTATION_PATH
        )
        if normal_rotation_filenames:
            return normal_rotation_filenames

    #
    # Von 00:00 bis 10:00 (oder wenn nichts anderes gefunden wurde) aus der
    # alten Rotation wählen.
    #
    return get_fresh_file_paths(recently_played_file_paths, OLD_ROTATION_PATH)


def main():
    try:
        with STATUS_FILE_PATH.open(encoding="utf-8") as file:
            recently_played_file_paths = list(map(Path, json.load(file)))
    except FileNotFoundError:
        recently_played_file_paths = []

    chosen_file_path = random.choice(
        choose_rotation(recently_played_file_paths)
    )
    recently_played_file_paths.append(chosen_file_path)
    
    with STATUS_FILE_PATH.open(encoding="utf-8") as file:
        json.dump(
            list(map(str, recently_played_file_paths[-MAX_RECENTLY_PLAYED:])),
            file,
        )

    print(chosen_file_path)


if __name__ == "__main__":
    main()