Only print full request time

src/controllers/slack_controller.py

@@ -1,36 +1,11 @@
-import time
-import socket
-import ssl
-import os
-
 from flask import jsonify
+from threading import Thread
 from services.slack_service import (
   process_event_payload
 )
-from threading import Thread
-
-def resolve_host(host):
-  start = time.time()
-  ip_address = socket.gethostbyname(host)
-  end = time.time()
-  return end - start
-
-def make_ssl_connection(host, port):
-  start = time.time()
-  sock = socket.create_connection((host, port))
-  ssl_sock = ssl.wrap_socket(sock)
-  end = time.time()
-  return end - start
 
 # POST /slack/events
 def post_event(request):
-    # host = "api.openai.com"
-    # resolution_time = resolve_host(host)
-    # print("It took %.2f seconds to resolve %s." % (resolution_time, host))
-    # port = 443
-    # connection_time = make_ssl_connection(host, port)
-    # print("It took %.2f seconds to make an SSL connection to %s." % (connection_time, host))
-
     if request.get("type") == "url_verification":
         return request.get("challenge")
 

src/index.py

@@ -1,5 +1,8 @@
+import logging
+import time
 import os
-from flask import Flask
+from logging import Logger
+from flask import Flask, g, request
 from slack_bolt.adapter.flask import SlackRequestHandler
 from routes.slack_router import slack_router
 from services.slack_service import slack_app
@@ -9,5 +12,25 @@
 flask_app = Flask("Haly")
 flask_app.register_blueprint(slack_router, url_prefix="/slack")
 
+##########################################
+# Config logger
+##########################################
+logging.basicConfig(level=logging.INFO)
+flask_app.logger: Logger
+
+@flask_app.before_request
+def before_request():
+    g.start_time = time.perf_counter()
+    g.route = request.path
+
+@flask_app.after_request
+def after_request(response):
+    end_time = time.perf_counter()
+    processing_time = end_time - g.start_time
+
+    flask_app.logger.info(f"Route: {g.route}, Request processed in {processing_time} seconds")
+
+    return response
+
 if __name__ == "__main__":
     flask_app.run(debug=True, host="0.0.0.0", port=int(os.environ.get("PORT", 8080)), threaded=True)

src/services/api_service.py

@@ -11,7 +11,6 @@
 SHARED_SECRET = os.environ["API_SHARED_SECRET"]
 
 
-@time_tracker
 def get_team_data(team_id):
     url = f"{BASE_URL}/api/organization/get_team_data/{team_id}"
     headers = {"X-Shared-Secret": SHARED_SECRET}
@@ -46,7 +45,6 @@ def revoke_token(team_id):
     return
 
 
-@time_tracker
 def increment_request_count(team_id):
     url = f"{BASE_URL}/api/slack/increment_request_count"
     headers = {

src/services/openai_service.py

@@ -15,7 +15,6 @@
 Users will sign their messages with their names, you won't"
 
 
-@time_tracker
 def run_completion(slack_messages, openai_key):
     openai.api_key = openai_key
     messages = [

src/services/slack_service.py

@@ -27,7 +27,6 @@ def send_message(channel: str, thread_ts: str, text: str, slack_bot_token: str):
     return response["ts"]
 
 
-@time_tracker
 def update_message(channel: str, thread_ts: str, ts: str, text: str, slack_bot_token: str):
     response = retry(
         lambda: slack_app.client.chat_update(
@@ -49,7 +48,6 @@ def delete_message(channel: str, ts: str, slack_bot_token: str):
     return response["ts"]
 
 
-@time_tracker
 def get_thread_messages(channel: str, thread_ts: str, slack_bot_token: str):
     try:
         return retry(
@@ -83,7 +81,6 @@ def find_user_by_id(user_id: str, slack_bot_token: str):
         print(e)
 
 
-@time_tracker
 def get_user_name(user_id: str, slack_bot_token: str):
     if user_id not in users_map:
         user = find_user_by_id(user_id, slack_bot_token)