XjfdZddlZddlmZddlZddlmZejjejje ddZ ee ejddZ eejd d Zejd d Zejd d ZejddZdedej(fdZedefdZedZy)a Database connection helper for the Scheduler service. Mirrors the PHP DBConnectionHelper: reads DB_MASTER_* credentials from the environment and creates per-request connections to the organisation-specific database whose name arrives in the X-DB-Name header. N)contextmanager) load_dotenvz..z.envDB_MASTER_HOSTz 127.0.0.1DB_MASTER_PORT3306DB_MASTER_USERNAMEDB_MASTER_PASSWORDDB_MASTER_DATABASE MYL_MASTERdatabasereturnc tjtttt |dtj jS)zCReturn a new pymysql connection to *database* on the master server.utf8mb4)hostportuserpasswordr charset cursorclass)pymysqlconnectrrrr cursors DictCursor)r s -/var/www/html/myl_app/scheduler-service/db.py_get_connectionrs2 ??   #OO.. db_namec#vKt|} ||jy#|jwxYww)z Context manager that yields a pymysql connection to the organisation database identified by *db_name*. The connection is closed on exit. N)rclose)rconns rget_dynamic_connectionr"(s. 7 #D   s 9$969c#~Ktt} ||jy#|jwxYww)zr Context manager that yields a pymysql connection to MYL_MASTER. Used for auth lookups (users table). N)rr r )r!s rget_master_connectionr$5s/ - .D   s=(=:=)__doc__os contextlibrrdotenvrpathjoindirname__file__ _ENV_PATHgetenvrintrrr r str Connectionrr"r$rrr3s % GGLL2D& A I+[9YRYY/89RYY3R8RYY3R8RYY3\B c g&8&8  C    r