1 files changed, 13 insertions, 5 deletions
diff --git a/service-type-database/build-db.in b/service-type-database/build-db.in
index 16a50e4..c27eb90 100755
--- a/service-type-database/build-db.in
+++ b/service-type-database/build-db.in
@@ -1,5 +1,5 @@
 #!@PYTHON@
-
+# -*-python-*-
 # $Id$
 
 # This file is part of avahi.
@@ -22,13 +22,21 @@
 import dbm, sys
 
 if len(sys.argv) > 1:
-    fn = sys.argv[1]
+    infn = sys.argv[1]
+else:
+    infn = "service-types"
+
+if len(sys.argv) > 2:
+    outfn = sys.argv[2]
 else:
-    fn = "service-types"
+    outfn = infn
+
+if outfn.endswith(".db"):
+    outfn = outfn[:-3]
 
-db = dbm.open(fn, "n")
+db = dbm.open(outfn, "n")
 
-for ln in file(fn, "r"):
+for ln in file(infn, "r"):
     ln = ln.strip(" \r\n\t")
     
     if ln == "" or ln.startswith("#"):