Add option to throttle responses

1 files changed, 5 insertions, 1 deletions

diff --git a/serve.py b/serve.py
index 433e740..2958968 100755
--- a/serve.py
+++ b/serve.py
@@ -1,6 +1,7 @@
 #!/usr/bin/env python
 
 import os
+import time
 import argparse
 from glob import glob
 from io import BytesIO
@@ -28,6 +29,10 @@ class Handler(BaseHTTPRequestHandler):
             self.send_error(404, msg)
             return
 
+        if os.getenv('GSX_THROTTLE'):
+            self.log_message('Throttling for %d' % os.getenv('GSX_THROTTLE'))
+            time.sleep(os.getenv('GSX_THROTTLE'))
+
         self.send_response(200)
         l = int(self.headers['Content-Length'])
         request = etree.parse(BytesIO(self.rfile.read(l)))
@@ -62,7 +67,6 @@ def validate_responses():
 
 if __name__ == '__main__':
     parser = argparse.ArgumentParser()
-    parser = argparse.ArgumentParser()
     parser.add_argument('-a', '--address',
         help='Address to host server on', default='localhost')
     parser.add_argument('-p', '--port',