Use sqlite for storage

src/Utils/__init__.py

@@ -1,15 +1,40 @@
 #!/usr/bin/env python3
 import io
-from os import path
-from posixpath import relpath
+import json
+import sqlite3
+from os import environ, makedirs, path
+from typing import Callable, Union
+from urllib.parse import urlparse
 
 import requests
 import wx
-from typing import Union, Callable
 
 SIZE = wx.Size(100, 68)
 MYPATH = path.dirname(path.abspath(__file__))
-SCREEN_WIDTH = 720 /2
+SCREEN_WIDTH = int(720 / 2)
+BASEPATH = path.join(str(environ.get("HOME")), '.config/cast')
+DB_FILE_NAME = 'cast.db'
+SUB_TABLE = 'subscriptions'
+
+
+def add_subscription(channel_id: str,
+                     name: str,
+                     basepath: str = BASEPATH,
+                     filename: str = DB_FILE_NAME) -> None:
+    fullpath = path.join(basepath, filename)
+    if not path.isdir(basepath):
+        makedirs(basepath)
+    con = sqlite3.connect(fullpath)
+    cur = con.cursor()
+    create_query: str = '''CREATE TABLE IF NOT EXISTS {}
+    (channel_id TEXT PRIMARY KEY, channel_name TEXT)'''.format(SUB_TABLE)
+    cur.execute(create_query)
+    con.commit()
+    upsert_query: str = '''INSERT INTO {} (channel_id, channel_name)
+    VALUES('{}',"{}") ON CONFLICT(channel_id) DO NOTHING'''.format(SUB_TABLE, channel_id, name)
+    cur.execute(upsert_query)
+    con.commit()
+
 
 def get_default_logo(providerid: str = 'default') -> wx.Bitmap:
     if providerid == 'SVT':
@@ -19,7 +44,36 @@ def get_default_logo(providerid: str = 'default') -> wx.Bitmap:
     else:
         return wx.Bitmap('{}/assets/Default.png'.format(MYPATH))
 
-def make_sized_button(parent_pnl: wx.Panel, bitmap_or_str: Union[wx.Bitmap,str], text: str, callback: Callable) -> wx.BoxSizer:
+
+def get_subscriptions(basepath: str = BASEPATH,
+                      filename: str = DB_FILE_NAME) -> list[tuple[str, str]]:
+    subscriptions = list()
+    fullpath = path.join(basepath, filename)
+    con = sqlite3.connect(fullpath)
+    cur = con.cursor()
+    select_query = '''SELECT * FROM {}'''.format(SUB_TABLE)
+    cur.execute(select_query)
+    for result in cur.fetchall():
+        print(result)
+        subscriptions.append(result)
+    return subscriptions
+
+
+def import_from_newpipe(filename) -> None:
+
+    if path.isfile(filename):
+        with open(filename, 'r') as subs:
+            sub_data = json.loads(subs.read())
+
+            for channel in sub_data['subscriptions']:
+                if channel['service_id'] == 0:
+                    channel_id = urlparse(channel['url']).path.split('/').pop()
+                    add_subscription(channel_id, channel['name'])
+
+
+def make_sized_button(parent_pnl: wx.Panel, bitmap_or_str: Union[wx.Bitmap,
+                                                                 str],
+                      text: str, callback: Callable) -> wx.BoxSizer:
     btn_sizer = wx.BoxSizer(wx.HORIZONTAL)
     if type(bitmap_or_str) == type(str):
         if bitmap_or_str.startswith('http'):  # type: ignore
@@ -34,8 +88,12 @@ def make_sized_button(parent_pnl: wx.Panel, bitmap_or_str: Union[wx.Bitmap,str],
     btn_logo.SetToolTip(text)
     btn_sizer.Add(btn_logo, 0, wx.BOTTOM | wx.EXPAND | wx.LEFT | wx.TOP, 1)
 
-        btn_text = wx.Button(parent_pnl, wx.ID_ANY, text, style=wx.BORDER_NONE | wx.BU_AUTODRAW)
-        btn_text.SetMinSize(wx.Size(SCREEN_WIDTH - SIZE.GetWidth(),SIZE.GetHeight()))
+    btn_text = wx.Button(parent_pnl,
+                         wx.ID_ANY,
+                         text,
+                         style=wx.BORDER_NONE | wx.BU_AUTODRAW)
+    btn_text.SetMinSize(
+        wx.Size(SCREEN_WIDTH - SIZE.GetWidth(), SIZE.GetHeight()))
     btn_text.SetToolTip(text)
     btn_sizer.Add(btn_text, 0, wx.BOTTOM | wx.RIGHT | wx.TOP | wx.EXPAND, 1)
     parent_pnl.Bind(wx.EVT_BUTTON, callback, btn_logo)
@@ -66,39 +124,44 @@ def make_bitmap_from_file(path, size: wx.Size = SIZE) -> wx.Bitmap:
     # image.Rescale(size.GetWidth(), size.GetHeight() )
     return wx.Bitmap(image)
 
+
 def resolve_svt_channel(svt_id: str) -> dict:
 
     channels = {
         "ch-barnkanalen": {
             "name":
             "Barnkanalen",
-            "thumbnail": make_bitmap_from_file('{}/assets/Barnkanalen.png'.format(MYPATH))
+            "thumbnail":
+            make_bitmap_from_file('{}/assets/Barnkanalen.png'.format(MYPATH))
         },
         "ch-svt1": {
-            "name":
-            "SVT 1",
-            "thumbnail": make_bitmap_from_file('{}/assets/SVT1.png'.format(MYPATH))
+            "name": "SVT 1",
+            "thumbnail":
+            make_bitmap_from_file('{}/assets/SVT1.png'.format(MYPATH))
         },
         "ch-svt2": {
-            "name":
-            "SVT 2",
-            "thumbnail": make_bitmap_from_file('{}/assets/SVT2.png'.format(MYPATH))
+            "name": "SVT 2",
+            "thumbnail":
+            make_bitmap_from_file('{}/assets/SVT2.png'.format(MYPATH))
         },
         "ch-svt24": {
             "name":
             "SVT 24",
-            "thumbnail": make_bitmap_from_file('{}/assets/SVT24.png'.format(MYPATH))
+            "thumbnail":
+            make_bitmap_from_file('{}/assets/SVT24.png'.format(MYPATH))
         },
         "ch-kunskapskanalen": {
             "name":
             "Kunskapskanalen",
-            "thumbnail": make_bitmap_from_file('{}/assets/Kunskapskanalen.png'.format(MYPATH))
+            "thumbnail":
+            make_bitmap_from_file(
+                '{}/assets/Kunskapskanalen.png'.format(MYPATH))
         },
         "feed": {
             "name": "Senaste program",
-            "thumbnail": make_bitmap_from_file('{}/assets/SVT.png'.format(MYPATH))
+            "thumbnail":
+            make_bitmap_from_file('{}/assets/SVT.png'.format(MYPATH))
         },
     }
 
     return channels[svt_id]
-

src/main.py

@@ -13,10 +13,10 @@ import wx.media
 
 from Channel import SVT, YouTube
 from ChannelProvider import ChannelProvider
-from Utils import make_sized_button
+from Utils import get_subscriptions, import_from_newpipe, make_sized_button
 
-WIDTH = 720 / 2
-HEIGHT = 1440 / 2
+WIDTH = int(720 / 2)
+HEIGHT = int(1440 / 2)
 BTN_HEIGHT = 40
 SPACER_HEIGHT = 10
 SCROLL_RATE = 5
@@ -91,15 +91,11 @@ class Cast(wx.Frame):
         subfile = 'yt_subs.json'
 
         if os.path.isfile(subfile):
-            with open(subfile, 'r') as subs:
-                janson = json.loads(subs.read())
-
-                for channel in janson['subscriptions']:
-                    if channel['service_id'] == 0:
-                        channel_id = urlparse(
-                            channel['url']).path.split('/').pop()
-                        youtube.append_channel(
-                            YouTube.YouTube(channel_id, channel['name']))
+            import_from_newpipe(subfile)
+            subscriptions = get_subscriptions()
+            for channel in subscriptions:
+                print(channel)
+                youtube.append_channel(YouTube.YouTube(channel[0], channel[1]))
 
         providers.append(youtube)
 
@@ -217,7 +213,7 @@ class Cast(wx.Frame):
         inner_sizer = wx.GridBagSizer()
         self.m_control = wx.media.MediaCtrl(
             self.m_panel,
-            size=(WIDTH, HEIGHT / 2),
+            size=(WIDTH, int(HEIGHT / 2)),
             style=wx.SIMPLE_BORDER,
             szBackend=wx.media.MEDIABACKEND_GSTREAMER,
         )