Updated Pipeline To Support Multi-OS & Added Support for PyPI Release

coldsofttech · Apr 13, 2024 · 034c8f8 · 034c8f8
1 parent 87b6835
commit 034c8f8
Show file tree

Hide file tree

Showing 5 changed files with 30 additions and 15 deletions.
diff --git a/src/__init__.py b/src/__init__.py
@@ -1,5 +1,6 @@
 import json
 import os
+import platform
 import sys
 import threading
 import tkinter as tk
@@ -459,9 +460,11 @@ def _open_license(self, event) -> None:
 
 
 def main():
-    if os.environ.get('DISPLAY', '') == '':
-        print('No display found. Using: 0.0')
-        os.environ.__setitem__('DISPLAY', ':0.0')
+    if platform.system().lower() == 'linux':
+        if platform.dist()[0].lower() == 'ubuntu':
+            if os.environ.get('DISPLAY', '') == '':
+                print('No display found. Using: 0.0')
+                os.environ.__setitem__('DISPLAY', ':0.0')
 
     root = tk.Tk()
     M3U8DownloaderUI(root)

diff --git a/tests/unit/test_aboutui.py b/tests/unit/test_aboutui.py
@@ -1,4 +1,5 @@
 import os
+import platform
 import tkinter as tk
 import unittest
 from unittest.mock import patch
@@ -12,9 +13,11 @@ class TestAboutUI(unittest.TestCase):
     """Unit test cases for AboutUI"""
 
     def setUp(self):
-        if os.environ.get('DISPLAY', '') == '':
-            print('No display found. Using: 0.0')
-            os.environ.__setitem__('DISPLAY', ':0.0')
+        if platform.system().lower() == 'linux':
+            if platform.dist()[0].lower() == 'ubuntu':
+                if os.environ.get('DISPLAY', '') == '':
+                    print('No display found. Using: 0.0')
+                    os.environ.__setitem__('DISPLAY', ':0.0')
 
         self.root = tk.Tk()
 

diff --git a/tests/unit/test_downloadthread.py b/tests/unit/test_downloadthread.py
@@ -1,5 +1,6 @@
 import json
 import os
+import platform
 import tkinter as tk
 import unittest
 
@@ -12,9 +13,11 @@ class TestDownloadThread(unittest.TestCase):
     """Unit test cases for DownloadThread class."""
 
     def setUp(self):
-        if os.environ.get('DISPLAY', '') == '':
-            print('No display found. Using: 0.0')
-            os.environ.__setitem__('DISPLAY', ':0.0')
+        if platform.system().lower() == 'linux':
+            if platform.dist()[0].lower() == 'ubuntu':
+                if os.environ.get('DISPLAY', '') == '':
+                    print('No display found. Using: 0.0')
+                    os.environ.__setitem__('DISPLAY', ':0.0')
 
         self.input_url = 'https://raw.githubusercontent.com/coldsofttech/pym3u8downloader/main/tests/files/index.m3u8'
         self.output_file = 'video.mp4'

diff --git a/tests/unit/test_m3u8downloaderui.py b/tests/unit/test_m3u8downloaderui.py
@@ -1,4 +1,5 @@
 import os
+import platform
 import tkinter as tk
 import unittest
 from unittest.mock import patch, MagicMock
@@ -12,9 +13,11 @@ class TestM3U8DownloaderUI(unittest.TestCase):
     """Unit test cases for M3U8DownloaderUI"""
 
     def setUp(self):
-        if os.environ.get('DISPLAY', '') == '':
-            print('No display found. Using: 0.0')
-            os.environ.__setitem__('DISPLAY', ':0.0')
+        if platform.system().lower() == 'linux':
+            if platform.dist()[0].lower() == 'ubuntu':
+                if os.environ.get('DISPLAY', '') == '':
+                    print('No display found. Using: 0.0')
+                    os.environ.__setitem__('DISPLAY', ':0.0')
 
         self.root = tk.Tk()
         self.source = M3U8DownloaderUI(self.root)

diff --git a/tests/unit/test_stdoutredirector.py b/tests/unit/test_stdoutredirector.py
@@ -1,4 +1,5 @@
 import os
+import platform
 import tkinter as tk
 import unittest
 
@@ -11,9 +12,11 @@ class TestStdoutRedirector(unittest.TestCase):
     """Unit test cases for StdoutRedirector"""
 
     def setUp(self):
-        if os.environ.get('DISPLAY', '') == '':
-            print('No display found. Using: 0.0')
-            os.environ.__setitem__('DISPLAY', ':0.0')
+        if platform.system().lower() == 'linux':
+            if platform.dist()[0].lower() == 'ubuntu':
+                if os.environ.get('DISPLAY', '') == '':
+                    print('No display found. Using: 0.0')
+                    os.environ.__setitem__('DISPLAY', ':0.0')
 
         self.root = tk.Tk()
         self.text_variable = tk.StringVar()