From bfa4b98a09f2f7bcb5139a9798562e052d6bed0f Mon Sep 17 00:00:00 2001
From: MrS0m30n3 <ytubedlg@gmail.com>
Date: Tue, 20 Sep 2016 19:48:14 +0300
Subject: [PATCH] Add DownloadItem & DownlaodPlaylist objects

---
 youtube_dl_gui/mainframe.py | 46 +++++++++++++++++++++++++++++++++++++
 1 file changed, 46 insertions(+)

diff --git a/youtube_dl_gui/mainframe.py b/youtube_dl_gui/mainframe.py
index ff974a5..f75fa08 100644
--- a/youtube_dl_gui/mainframe.py
+++ b/youtube_dl_gui/mainframe.py
@@ -931,3 +931,49 @@ class ExtComboBox(wx.ComboBox):
     def LoadMultiple(self, items_list):
         for item in items_list:
             self.Append(item)
+
+
+class DownloadItem(object):
+
+    def __init__(self, url, options, filename="", extension="", path="", filesize=0.0, playlist_index=-1):
+        self.url = url
+        self.path = path
+        self.options = options
+        self.filename = filename
+        self.filesize = filesize
+        self.extension = extension
+        self.playlist_index = playlist_index
+
+        self._progress_stats = {
+            "filesize": "",
+            "percent": "",
+            "status": "",
+            "speed": "",
+            "eta": ""
+        }
+
+        self.object_id = id(self)
+
+        def get_eta(self):
+            return self._progress_stats["eta"]
+
+        def get_speed(self):
+            return self._progress_stats["speed"]
+
+        def get_status(self):
+            return self._progress_stats["status"]
+
+        def get_percentage(self):
+            return self._progress_stats["percent"]
+
+        def get_absolute_path(self):
+            return os.path.join(self.path, self.filename, self.extension)
+
+
+class DownloadPlaylist(object):
+
+    # TODO feauture use
+
+    def __init__(self, size, items):
+        self.size = size
+        self.items = items