Allow including source filename in auto-generated header.

diff --git a/asn1ate/support/pygen.py b/asn1ate/support/pygen.py
index 91388a6..a47e75f 100644 (file)
--- a/asn1ate/support/pygen.py
+++ b/asn1ate/support/pygen.py
@@ -33,8 +33,12 @@ except ImportError:
 from datetime import datetime
 
 
-def auto_generated_header():
-    return '# Auto-generated by asn1ate on %s' % datetime.now()
+def auto_generated_header(source_filename=None):
+    prefix = '# Auto-generated'
+    if source_filename:
+        prefix += ' from %s' % source_filename
+
+    return '%s by asn1ate on %s' % (prefix, datetime.now())
 
 
 class NullBackend(object):

diff --git a/asn1ate/test.py b/asn1ate/test.py
index 66a663b..c75da47 100644 (file)
--- a/asn1ate/test.py
+++ b/asn1ate/test.py
@@ -60,7 +60,7 @@ def main():
 
     if args.gen:
         for module in modules:
-            print(pygen.auto_generated_header())
+            print(pygen.auto_generated_header(args.file))
             pyasn1gen.generate_pyasn1(module, sys.stdout)
 
     return 0