session for renew_key

handlers/coupons.py

@@ -229,6 +229,7 @@ async def handle_key_extension(
             client_id=client_id,
             new_expiry_time=new_expiry,
             total_gb=0,
+            session=session
         )
         await update_key_expiry(session, client_id, new_expiry)
         await update_coupon_usage_count(session, coupon.id)

handlers/notifications/general_notifications.py

@@ -572,7 +572,7 @@ async def process_auto_renew_or_notify(
         )
 
         await renew_key_in_cluster(
-            server_id, email, client_id, int(new_expiry_time), total_gb
+            server_id, email, client_id, int(new_expiry_time), total_gb, session=conn
         )
         await update_balance(conn, tg_id, -renewal_cost)
         await update_key_expiry(conn, client_id, int(new_expiry_time))