From 68e0ec663e89742fd0f75808e14579388547ab9d Mon Sep 17 00:00:00 2001 From: headroom Date: Mon, 13 Jan 2025 15:45:56 +0000 Subject: [PATCH] Add song_kissFM_insert.py --- song_kissFM_insert.py | 85 +++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 85 insertions(+) create mode 100644 song_kissFM_insert.py diff --git a/song_kissFM_insert.py b/song_kissFM_insert.py new file mode 100644 index 0000000..08a053e --- /dev/null +++ b/song_kissFM_insert.py @@ -0,0 +1,85 @@ +import requests +from bs4 import BeautifulSoup +from datetime import datetime, timedelta +import base64 +import argparse +import random +import psycopg2 # Afegit per a la connexió amb PostgreSQL + +# URL de l'emissora amb entropia per evitar la memòria cau +BASE_URL = "https://emisora.org.es/kiss-fm/" +URL_WITH_ENTROPY = f"{BASE_URL}?rand={random.randint(0, 100000)}" + +# Funció per a connectar a la base de dades PostgreSQL +def connect_db(): + return psycopg2.connect( + dbname="playlist", + user="kiss", + password="PASSWORD", # Substitueix per la teva contrasenya real + host="localhost" + ) + +# Procésa el darrer element `li` de `data-playlist-previous-songs` +def fetch_last_playlist_song(): + headers = { + 'Cache-Control': 'no-cache', + 'Pragma': 'no-cache', + 'User-Agent': 'Mozilla/5.0 (Windows NT 5.1; rv:40.0) Gecko/20100101 Firefox/40.0' + } + + response = requests.get(URL_WITH_ENTROPY, headers=headers) + if response.status_code == 200: + soup = BeautifulSoup(response.content, 'html.parser') + last_li = soup.select_one('ul[data-playlist-previous-songs] li:last-child') + + if last_li: + timestamp = datetime.now().strftime("%Y-%m-%d %H:%M:%S") + title = last_li.get('title', 'N/A') + artist = last_li.find('span', class_='playlist__artist-name').text + song = last_li.find('span', class_='playlist__song-name').text + img_tag = last_li.find('img') + img_src = img_tag['src'] if img_tag else '' + + # Obtenim les dades de la imatge i les convertim en base64 + if img_src: + img_response = requests.get(img_src) + if img_response.status_code == 200: + img_data = base64.b64encode(img_response.content).decode('utf-8') + img_md = f"![Imatge](data:image/jpeg;base64,{img_data})" + else: + img_md = "No disponible" + else: + img_md = "No disponible" + + # Inserir a la base de dades + conn = connect_db() + cursor = conn.cursor() + cursor.execute( + "INSERT INTO kissfm (timestamp, title, artist, song, img_md) VALUES (%s, %s, %s, %s, %s)", + (timestamp, title, artist, song, img_md) + ) + conn.commit() + cursor.close() + conn.close() + +# Genera un resum diari de la música detectada +def generate_daily_summary(): + yesterday = (datetime.now() - timedelta(days=1)).strftime("%Y-%m-%d") + + # Aquí s'hauria d'implementar la lògica per llegir el fitxer de resum diari + # i desar les dades a la base de dades com en fetch_last_playlist_song() + +# Analitza els arguments de la línia de comandes +def main(): + parser = argparse.ArgumentParser() + parser.add_argument('--resum', action='store_true', help="Generar el resum diari") + args = parser.parse_args() + + if args.resum: + generate_daily_summary() # Aquí cridem a la funció que hauries d'implementar + else: + fetch_last_playlist_song() + + +if __name__ == "__main__": + main() \ No newline at end of file