Add song_kissFM_insert.py

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()