import os
from http.server import BaseHTTPRequestHandler, HTTPServer
import pyodbc
# Database connection parameters
DB_HOST = os.getenv('DB_HOST')
DB_PORT = os.getenv('DB_PORT')
DB_NAME = os.getenv('DB_NAME')
DB_USER = os.getenv('DB_USER')
DB_PASSWORD = os.getenv('DB_PASSWORD')
# Connection string
DATABASE_URI = f"DRIVER={{ODBC Driver 18 for SQL Server}};SERVER={DB_HOST},{DB_PORT};DATABASE={DB_NAME};UID={DB_USER};PWD={DB_PASSWORD};Encrypt=no;"
class RequestHandler(BaseHTTPRequestHandler):
def do_GET(self):
connection = None # Initialize connection to None
try:
# Connect to the database
connection = pyodbc.connect(DATABASE_URI)
cursor = connection.cursor()
cursor.execute("SELECT @@VERSION;")
db_version = cursor.fetchone()
response = f"Connection successful, MSSQL version: {db_version[0]}"
except Exception as error:
response = f"Error connecting without pooling: {error}"
finally:
if connection:
cursor.close()
connection.close()
# Send response to client
self.send_response(200)
self.send_header("Content-type", "text/html")
self.end_headers()
self.wfile.write(f"<html><body><h1>{response}</h1></body></html>".encode("utf-8"))
def run(server_class=HTTPServer, handler_class=RequestHandler):
server_address = ('', 8080) # Serve on port 8080
httpd = server_class(server_address, handler_class)
print("Server running on port 8080")
httpd.serve_forever()
if __name__ == "__main__":
run()
