Use the same magics setup code also in test-render. Refs: #2

arkimaps

@@ -205,6 +205,8 @@ def main():
 
     # Tell magics where it shuold take its default styles from
     os.environ["MAGICS_STYLE_PATH"] = args.styles
+    # Tell magics not to print noisy banners
+    os.environ["MAGPLUS_QUIET"] = "1"
 
     with workdir() as root:
         # Dispatch input products into a work directory

test-render

@@ -52,16 +52,18 @@ def render(fname):
 
 
 def main():
+    default_styles = os.path.abspath(os.path.join(os.path.dirname(sys.argv[0]), "styles"))
+    if not os.path.isdir(default_styles):
+        default_styles = "/usr/share/magics/styles/ecmwf"
+
     parser = argparse.ArgumentParser(description="Render model maps")
     parser.add_argument("--verbose", "-v", action="store_true", help="verbose output")
     parser.add_argument("--debug", action="store_true", help="debug output")
+    parser.add_argument("--styles", metavar="dir", action="store", default=default_styles, help="styles directory")
     parser.add_argument("files", nargs="+", help="files to render")
 
     args = parser.parse_args()
 
-    # Tell magics not to print noisy banners
-    os.environ["MAGPLUS_QUIET"] = "1"
-
     # Setup logging
     FORMAT = "%(asctime)-15s %(levelname)s %(message)s"
     if args.debug:
@@ -71,6 +73,11 @@ def main():
     else:
         logging.basicConfig(level=logging.WARN, stream=sys.stderr, format=FORMAT)
 
+    # Tell magics where it shuold take its default styles from
+    os.environ["MAGICS_STYLE_PATH"] = args.styles
+    # Tell magics not to print noisy banners
+    os.environ["MAGPLUS_QUIET"] = "1"
+
     for fname in args.files:
         render(fname)