Add `get_password`

Make a function that will get a password string given a shell command.
This allows us to pass a command to the program instead of a clear text
password.

1 files changed, 7 insertions, 0 deletions

diff --git a/harvester_submit_week_for_approval.py b/harvester_submit_week_for_approval.py
index 10d8dd9..77e0665 100644
--- a/harvester_submit_week_for_approval.py
+++ b/harvester_submit_week_for_approval.py
@@ -2,6 +2,7 @@
 
 import argparse
 from datetime import datetime, timedelta
+import subprocess
 
 from selenium.webdriver import Firefox
 from selenium.webdriver.common.by import By
@@ -80,6 +81,12 @@ def submit_week_for_approval(driver, wait):
 
     return driver
 
+def get_password(command):
+    result = subprocess.run(['sh', '-c', command], stdout=subprocess.PIPE)
+
+    return str(result.stdout, 'utf-8').rstrip()
+
+
 if __name__ == "__main__":
     parser = argparse.ArgumentParser(
         description='Submit the most recent Harvest timesheet for approval.')