Merge: Nitpretty: make the tool more customizable

[nit.git] / src / nitpretty.nit

diff --git a/src/nitpretty.nit b/src/nitpretty.nit
index 1d7641d..9dd723e 100644 (file)
--- a/src/nitpretty.nit
+++ b/src/nitpretty.nit
@@ -108,12 +108,11 @@ if toolcontext.opt_skip_empty.value then
 end
 
 for mmodule in mmodules do
-       if not mbuilder.mmodule2nmodule.has_key(mmodule) then
+       var nmodule = mbuilder.mmodule2node(mmodule)
+       if nmodule == null then
                print " Error: no source file for module {mmodule}"
                return
        end
-
-       var nmodule = mbuilder.mmodule2nmodule[mmodule]
        var file = "{dir}/{mmodule.name}.nit"
        var tpl = v.pretty_nmodule(nmodule)
        tpl.write_to_file file