From: Guillaume Pellerin <guillaume.pellerin@free.fr>
Date: Tue, 23 Apr 2024 16:06:07 +0000 (+0200)
Subject: pass namespace to init
X-Git-Url: https://git.parisson.com/?a=commitdiff_plain;h=72fb04975978b8f1131bf1b6417cdf1eec243a47;p=telecaster-server.git

pass namespace to init
---

diff --git a/bin/mastering/mastering.py b/bin/mastering/mastering.py
index c0eefb7..94456c2 100755
--- a/bin/mastering/mastering.py
+++ b/bin/mastering/mastering.py
@@ -40,44 +40,38 @@ class TeleCasterMastering(object):
     date_limit = datetime.datetime(year=2024, month=4, day=19)
     tmp_dir = "/tmp/"
 
-    def __init__(self, root_dir,
-            log_file='/tmp/telecaster.log',
-            nvidia_mode=False,
-            vaapi_mode=False,
-            auto_offset_mode=False,
-            dry_run=False,
-            force_mode=False,
-            verbose_mode=False,
-            input_formats=None,
-            output_formats=None,
-            ):
-
-        self.log_file = log_file
-        self.logger = Logger(self.log_file)
-        self.root_dir = root_dir
-
-        self.nvidia_mode = nvidia_mode
+    def __init__(self, args):
+
+        if args.log:
+            self.log = args.log
+        else:
+            self.log = '/tmp/telecaster.log'
+
+        self.logger = Logger(self.log)
+        self.root_dir = args.dir
+
+        self.nvidia_mode = args.nvidia
         if self.nvidia_mode:
             for f in self.nvidia_formats:
                 self.dest_formats[f] = self.nvidia_formats[f]
 
-        self.vaapi_mode = vaapi_mode
+        self.vaapi_mode = args.vaapi
         if self.vaapi_mode:
             for f in self.vaapi_formats:
                 self.dest_formats[f] = self.vaapi_formats[f]
 
-        self.auto_offset_mode = auto_offset_mode
-        self.dry_run = dry_run
-        self.force_mode = force_mode
-        self.verbose_mode = verbose_mode
+        self.auto_offset_mode = args.auto_offset
+        self.dry_run = args.dry_run
+        self.force_mode = args.force
+        self.verbose_mode = args.verbose
 
-        if input_formats:
-            self.input_formats = input_formats
+        if args.input_formats:
+            self.input_formats = args.input_formats
         else:
             self.input_formats = self.source_formats
 
-        if output_formats:
-            self.output_formats = output_formats
+        if args.output_formats:
+            self.output_formats = args.output_formats
         else:
             self.output_formats = self.dest_formats
 
@@ -203,8 +197,8 @@ class TeleCasterMastering(object):
 
 def main():
     parser = argparse.ArgumentParser()
-    parser.add_argument('-d', '--dir', metavar='directory', type=str, help='media directory')
-    parser.add_argument('-l', '--log', metavar='logfile', type=str, help='log file', default='/tmp/telecaster.log')
+    parser.add_argument('-d', '--dir', type=str, help='media directory')
+    parser.add_argument('-l', '--log', type=str, help='log file', default='/tmp/telecaster.log')
     parser.add_argument('-nv', '--nvidia', help='use GPU nvidia encoder', action="store_true")
     parser.add_argument('-va', '--vaapi', help='use VAAPI encoder', action="store_true")
     parser.add_argument('-ao', '--auto_offset', help='add auto offset based on audio correlation', action="store_true")
@@ -215,19 +209,7 @@ def main():
     parser.add_argument('-o','--output_formats', nargs='+', help='Required output formats', required=True)
 
     args = parser.parse_args()
-
-    t = TeleCasterMastering(
-        args.dir,
-        log_file=args.log,
-        nvidia_mode=args.nvidia,
-        vaapi_mode=args.vaapi,
-        auto_offset_mode=args.auto_offset,
-        dry_run=args.dry_run,
-        verbose_mode=args.verbose,
-        input_formats=args.input_formats,
-        output_formats=args.output_formats,
-        )
-
+    t = TeleCasterMastering(args)
     t.run()