From 0c75a2896a9fd530292a5f0b71ea22b284167f35 Mon Sep 17 00:00:00 2001
From: JKuijperM <jkmontero17@gmail.com>
Date: Sat, 15 May 2021 10:20:58 +0200
Subject: [PATCH] Added the main code

---
 main.py | 36 ++++++++++++++++++++++++++++++++++++
 1 file changed, 36 insertions(+)
 create mode 100644 main.py

diff --git a/main.py b/main.py
new file mode 100644
index 0000000..787b50d
--- /dev/null
+++ b/main.py
@@ -0,0 +1,36 @@
+import csv
+import json
+import pandas as pd
+from pathlib import Path
+from datetime import datetime
+
+# csv_path = Path(r'D:\DESCARGAS\tviso-collection.csv')
+# df = pd.read_csv(csv_path, sep=';')
+
+json_path = Path('data/tviso-collection.json')
+with open(json_path, encoding='utf-8') as json_file:
+    data = json.load(json_file)
+
+# col = ['imdbID,Title,Rating10,WatchedDate']
+col = ['imdbID', 'Title', 'Rating10', 'WatchedDate']
+df = pd.DataFrame(columns=col)
+
+for line in data:
+    if line['status'] == 'watched' and line['type'] == 2:
+        title = line['title']
+        imdb = line['imdb']
+        rating = line['rating'] if line['rating'] is not None else ''
+        try:
+            watchedDate = line['checkedDate']
+        except:
+            watchedDate = ''
+
+        if watchedDate != '':
+            dt = datetime.strptime(watchedDate, '%Y-%m-%dT%H:%M:%S+%f:00')
+            watchedDate = dt.strftime('%Y-%m-%d')
+
+        # row_data = '{},{},{},{}'.format(imdb, title, rating, watchedDate)
+        row_data = [imdb, title, rating, watchedDate]
+        df.loc[len(df) + 1] = row_data
+
+df.to_csv('test.csv', index=False, encoding='utf-8')