r.out.kde: fix lazy import of 'PIL' module (#387)

OSGeo · Jan 2, 2021 · 69d743e · 69d743e
1 parent e1a1d99
commit 69d743e
Showing 1 changed file with 6 additions and 5 deletions.
diff --git a/grass7/raster/r.out.kde/r.out.kde.py b/grass7/raster/r.out.kde/r.out.kde.py
@@ -62,6 +62,12 @@ def cleanup():
 
 
 def main(rinput, background, output, method):
+    try:
+        from PIL import Image
+    except ImportError:
+        gscript.fatal("Cannot import PIL."
+                      " Please install the Python pillow package.")
+
     if '@' in rinput:
         rinput = rinput.split('@')[0]
     suffix = '_' + os.path.basename(gscript.tempfile(False))
@@ -125,11 +131,6 @@ def scale(cmin, cmax, intens, method):
 
 
 if __name__ == "__main__":
-    try:
-        from PIL import Image
-    except ImportError:
-        gscript.fatal("Cannot import PIL."
-                      " Please install the Python pillow package.")
     options, flags = gscript.parser()
     rinput = options['input']
     bg = options['background']