Implement single host as argument

ssh_terminfo.py

@@ -1,6 +1,7 @@
 #!/bin/python
 
 import os
+import sys
 import subprocess
 
 ssh_config = os.path.expanduser('~') + "/.ssh/config"
@@ -10,6 +11,14 @@ with open(ssh_config) as file:
 
 #ping -c 1 -w 1
 
+if sys.argv[1] != None:
+    ssh_host = sys.argv[1]
+    command = f"infocmp | ssh {ssh_host} 'tic -x /dev/stdin'"
+    return_value = subprocess.call(command, shell=True, stdout=open(os.devnull, "w"), stderr=subprocess.STDOUT)
+    if return_value == 1:
+        print(f"Could not send terminfo to {ssh_host}")
+
+else:
     for l in file_content:
         if "host " in l:
             l_index = file_content.index(l)