Move logic to separate function normalize_autorefresh.

ycdl/objects.py

@@ -48,6 +48,16 @@ class Channel(Base):
     def __repr__(self):
         return f'Channel:{self.id}'
 
+    @staticmethod
+    def normalize_autorefresh(autorefresh):
+        if isinstance(autorefresh, (str, int)):
+            autorefresh = stringtools.truthystring(autorefresh, none_set={})
+
+        if not isinstance(autorefresh, bool):
+            raise TypeError(f'autorefresh should be a boolean, not {autorefresh}.')
+
+        return autorefresh
+
     @staticmethod
     def normalize_download_directory(
             download_directory,
@@ -209,13 +219,7 @@ class Channel(Base):
             self.ycdldb.commit()
 
     def set_autorefresh(self, autorefresh, commit=True):
-        if isinstance(autorefresh, int):
+        autorefresh = self.normalize_autorefresh(autorefresh)
-            if autorefresh not in {0, 1}:
-                raise ValueError(f'autorefresh should be a boolean, not {autorefresh}.')
-            autorefresh = bool(autorefresh)
-
-        if not isinstance(autorefresh, bool):
-            raise TypeError(f'autorefresh should be a boolean, not {autorefresh}.')
 
         pairs = {
             'id': self.id,