diff --git a/websitemodule.py b/websitemodule.py index a4bdc4a..2ed9448 100644 --- a/websitemodule.py +++ b/websitemodule.py @@ -1,7 +1,8 @@ import os import sqlite3 -import requests import datetime +# For now +import flask def read_music_database(): conn = sqlite3.connect("assets/localwrapped/music.db") @@ -12,57 +13,214 @@ def read_music_database(): return music_list def read_count_database(year: int): - if year == datetime.datetime.now().year: - conn = sqlite3.connect(f'assets/localwrapped/count.db') - c = conn.cursor() - c.execute('SELECT * FROM count') - count_list = c.fetchall() - conn.close() - return count_list - else: - conn = sqlite3.connect(f'assets/localwrapped/count-{year}.db') - c = conn.cursor() - c.execute('SELECT * FROM count') - count_list = c.fetchall() - conn.close() - return count_list + db_path = f'assets/localwrapped/count-{year}.db' if year != datetime.datetime.now().year else 'assets/localwrapped/count.db' + if not os.path.exists(db_path): + return [] + + conn = sqlite3.connect(db_path) + c = conn.cursor() + c.execute('SELECT * FROM count') + count_list = c.fetchall() + conn.close() + return count_list + +def get_available_years(): + years = [] + for file in os.listdir('assets/localwrapped'): + if file.startswith('count') and file.endswith('.db'): + year = file.split('-')[-1].split('.')[0] + if year.isdigit(): + years.append(int(year)) + return sorted(years, reverse=True) def total_play_time(year: int): total_time = 0 - for countvar in read_count_database(year): - for musicvar in read_music_database(): - if countvar[0] == musicvar[0]: - total_time += float(musicvar[-1]) * countvar[1] + music_data = {row[0]: row for row in read_music_database()} + + for count_entry in read_count_database(year): + if count_entry[0] in music_data: + total_time += float(music_data[count_entry[0]][7]) * count_entry[1] + return total_time def generatehtmlcode(year: int = datetime.datetime.now().year): - html = '\n\n' - html += '\n\n\n\n' - html += '\n\n' - html += 'Halhadus\' Local Wrapped\n\n' - html += '\n\n' - html += '\n\n' - html += '\n' - html += '

Halhadus\' Local Wrapped

\n' - html += 'Source Code\n' - html += '

Hi guys. I am too lazy to add links to song names. To find which song is it, check this link.

\n' - html += '

' + str(year) + '

\n' - html += '
\n\n
\n' - html += '

Total Play Time: ' + str(int(total_play_time(datetime.datetime.now().year)/60)) + ' minutes

\n' - if year == datetime.datetime.now().year: - html += '

Last Updated: ' + datetime.datetime.utcfromtimestamp(os.path.getmtime('assets/localwrapped/count.db')).strftime('%Y-%m-%d %H:%M:%S') + ' (UTC +0, I hope.)

\n'; - else: - html += '

Last Updated: ' + datetime.datetime.utcfromtimestamp(os.path.getmtime(f'assets/localwrapped/count-{year}.db')).strftime('%Y-%m-%d %H:%M:%S') + ' (UTC +0, I hope.)

\n' - html += '

Most Played Musics

\n' - html += '
    \n' - count_list_formatted = [] - for music in read_count_database(year): - music_name = music[0] - count = music[1] - count_list_formatted.append([music_name, count]) - count_list_formatted.sort(key=lambda x: x[1], reverse=True) - for music in count_list_formatted: - html += '
  1. ' + music[0] + ' | ' + str(music[1]) + ' times
    2. \n' - html += '
\n' - html += '\n' - return html + available_years = get_available_years() + current_year = datetime.datetime.now().year + selected_year = year if year in available_years else current_year + + html = ''' + + + + + + Halhadus' Local Wrapped + + + + + + + + + + +
+

🎵 Halhadus' Local Wrapped

+ ← Main Page +
+ 📁 Source Codes +
+ +
+ {% for y in available_years %} + + {{ y }} + + {% endfor %} +
+ +
+

📊 Statistics for {{ selected_year }}

+

⏳ Total Play Time: {{ total_time }} minutes

+

🕒 Last Updated: {{ last_updated }}

+
+ +

🎶 Most Played Tracks

+ {% for music in top_tracks %} +
+ {{ music.name }} cover +
+

{{ music.name }}

+

▶ Played {{ music.count }} times

+
+ + YouTube + + + YT Music + +
+
+
+ {% endfor %} +
+ +''' + + # Verileri hazırla + music_data = {row[0]: row for row in read_music_database()} + count_data = read_count_database(selected_year) + + top_tracks = [] + for count_entry in count_data: + if count_entry[0] in music_data: + video_id = music_data[count_entry[0]][1] + top_tracks.append({ + 'name': count_entry[0], + 'count': count_entry[1], + 'video_id': video_id + }) + + top_tracks.sort(key=lambda x: x['count'], reverse=True) + + total_time = int(total_play_time(selected_year) / 60) + + db_path = f'assets/localwrapped/count-{selected_year}.db' if selected_year != current_year else 'assets/localwrapped/count.db' + last_updated = datetime.datetime.utcfromtimestamp(os.path.getmtime(db_path)).strftime('%Y-%m-%d %H:%M UTC') + + return flask.render_template_string(html, top_tracks=top_tracks, total_time=total_time, last_updated=last_updated, available_years=available_years, selected_year=selected_year) diff --git a/websitemodule.py.old b/websitemodule.py.old new file mode 100644 index 0000000..a4bdc4a --- /dev/null +++ b/websitemodule.py.old @@ -0,0 +1,68 @@ +import os +import sqlite3 +import requests +import datetime + +def read_music_database(): + conn = sqlite3.connect("assets/localwrapped/music.db") + c = conn.cursor() + c.execute('SELECT * FROM music') + music_list = c.fetchall() + conn.close() + return music_list + +def read_count_database(year: int): + if year == datetime.datetime.now().year: + conn = sqlite3.connect(f'assets/localwrapped/count.db') + c = conn.cursor() + c.execute('SELECT * FROM count') + count_list = c.fetchall() + conn.close() + return count_list + else: + conn = sqlite3.connect(f'assets/localwrapped/count-{year}.db') + c = conn.cursor() + c.execute('SELECT * FROM count') + count_list = c.fetchall() + conn.close() + return count_list + +def total_play_time(year: int): + total_time = 0 + for countvar in read_count_database(year): + for musicvar in read_music_database(): + if countvar[0] == musicvar[0]: + total_time += float(musicvar[-1]) * countvar[1] + return total_time + +def generatehtmlcode(year: int = datetime.datetime.now().year): + html = '\n\n' + html += '\n\n\n\n' + html += '\n\n' + html += 'Halhadus\' Local Wrapped\n\n' + html += '\n\n' + html += '\n\n' + html += '\n' + html += '

Halhadus\' Local Wrapped

\n' + html += 'Source Code\n' + html += '

Hi guys. I am too lazy to add links to song names. To find which song is it, check this link.

\n' + html += '

' + str(year) + '

\n' + html += '
\n\n
\n' + html += '

Total Play Time: ' + str(int(total_play_time(datetime.datetime.now().year)/60)) + ' minutes

\n' + if year == datetime.datetime.now().year: + html += '

Last Updated: ' + datetime.datetime.utcfromtimestamp(os.path.getmtime('assets/localwrapped/count.db')).strftime('%Y-%m-%d %H:%M:%S') + ' (UTC +0, I hope.)

\n'; + else: + html += '

Last Updated: ' + datetime.datetime.utcfromtimestamp(os.path.getmtime(f'assets/localwrapped/count-{year}.db')).strftime('%Y-%m-%d %H:%M:%S') + ' (UTC +0, I hope.)

\n' + html += '

Most Played Musics

\n' + html += '
    \n' + count_list_formatted = [] + for music in read_count_database(year): + music_name = music[0] + count = music[1] + count_list_formatted.append([music_name, count]) + count_list_formatted.sort(key=lambda x: x[1], reverse=True) + for music in count_list_formatted: + html += '
  1. ' + music[0] + ' | ' + str(music[1]) + ' times
    2. \n' + html += '
\n' + html += '\n' + return html