Support running local-cname on containers

See #4
hjacobs · Jun 15, 2020 · 173064c · 173064c
1 parent fc1a59a
commit 173064c
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/local_cname/cli.py b/local_cname/cli.py
@@ -1,6 +1,7 @@
 import argparse
 import socket
 import time
+import os
 from pathlib import Path
 
 from clickclick import Action, info
@@ -49,4 +50,9 @@ def main():
  # ignore, do not print stacktrace
  pass
  finally:
- backup_file.rename(hosts_file)
+ try:
+ backup_file.rename(hosts_file)
+ except OSError:
+ with hosts_file.open('w') as fd:
+ fd.write(old_contents)
+ os.remove(backup_file)