Allow setting of profile filename via OBNAM_PROFILE.

1 files changed, 5 insertions, 2 deletions

diff --git a/obnam b/obnam
index 4582304f..71a74e81 100755
--- a/obnam
+++ b/obnam
@@ -39,8 +39,11 @@ def main():
 
 
 if __name__ == '__main__':
-    if os.environ.get('OBNAM_PROFILE') == 'yes':
+    profname = os.environ.get('OBNAM_PROFILE')
+    if profname:
         import cProfile
-        cProfile.run('main()', 'obnam.prof')
+        if profname == 'yes':
+            profname = 'obnam.prof'
+        cProfile.run('main()', profname)
     else:
         main()