Making windows happy

1 files changed, 7 insertions, 0 deletions

diff --git a/mitmproxy/addons/command_history.py b/mitmproxy/addons/command_history.py
index c66af2be..8c1ef052 100644
--- a/mitmproxy/addons/command_history.py
+++ b/mitmproxy/addons/command_history.py
@@ -1,6 +1,7 @@
 import collections
 import os
 import typing
+import atexit
 
 from mitmproxy import command
 from mitmproxy import ctx
@@ -28,6 +29,12 @@ class CommandHistory:
         for l in _history_lines:
             self.add_command(l.strip())
 
+        atexit.register(self.cleanup)
+
+    def cleanup(self):
+        if self.command_history_file:
+            self.command_history_file.close()
+
     @property
     def last_filtered_index(self):
         return len(self.filtered_commands) - 1