local development use port 2200HEADmaster

1 files changed, 7 insertions, 2 deletions

diff --git a/log_ssh_passwords.py b/log_ssh_passwords.py
index 1ad7c97..5d2a38b 100644
--- a/log_ssh_passwords.py
+++ b/log_ssh_passwords.py
@@ -1,6 +1,7 @@
 #!/opt/public_ssh_password_stats/log_ssh_passwords/venv/bin/python3
 import socket
 import time
+import os
 
 import paramiko
 import db_handler
@@ -38,8 +39,12 @@ def ssh_thread(client, address, port):
 
 # Create an SSH server
 host_key = paramiko.RSAKey.from_private_key_file("id_rsa")  # paramiko.RSAKey.generate(2048)
-server = ''
-port = 2200
+if os.environ.get('DEBUG') == '1':  # Local development
+    server = '127.0.0.1'
+    port = 2200
+else:  # Production
+    server = ''
+    port = 22
 
 # Start the server
 server_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)