{ "cells": [ { "cell_type": "code", "execution_count": 2, "id": "e706cfb0-2234-4c4c-95d8-d1968f656aa0", "metadata": { "tags": [] }, "outputs": [], "source": [ "import pandas as pd\n", "\n", "query = \"\"\"\n", " select final.CUSTOMER_NUMBER_main as Focal_id,\n", " CAST(final.Cash_deposit_total AS DECIMAL(18, 2)) AS Cash_deposit_total,\n", " final.Cash_deposit_count,\n", " final.SEGMENT,\n", " final.RISK,\n", " final.SAR_FLAG\n", "from \n", "(\n", " (\n", " select subquery.CUSTOMER_NUMBER_1 as CUSTOMER_NUMBER_main,\n", " subquery.Cash_deposit_total,\n", " subquery.Cash_deposit_count\n", " from \n", " (\n", " select customer_number as CUSTOMER_NUMBER_1, \n", " sum(transaction_amount) as Cash_deposit_total, \n", " count(*) as Cash_deposit_count\n", " from \n", " (\n", " select * \n", " from {trans_data} trans_table \n", " left join {acc_data} acc_table\n", " on trans_table.benef_account_number = acc_table.account_number\n", " ) trans\n", " where account_number not in ('None')\n", " and transaction_desc = 'CASH RELATED TRANSACTION'\n", " group by customer_number\n", " ) subquery\n", " ) main \n", " left join \n", " (\n", " select cd.CUSTOMER_NUMBER_3 as CUSTOMER_NUMBER_cust,\n", " cd.SEGMENT,\n", " cd.RISK,\n", " case\n", " when ad.SAR_FLAG is NULL then 'N'\n", " else ad.SAR_FLAG\n", " end as SAR_FLAG \n", " from\n", " (\n", " select customer_number as CUSTOMER_NUMBER_3, \n", " business_segment as SEGMENT,\n", " case\n", " when RISK_CLASSIFICATION = 1 then 'Low Risk'\n", " when RISK_CLASSIFICATION = 2 then 'Medium Risk'\n", " when RISK_CLASSIFICATION = 3 then 'High Risk'\n", " else 'Unknown Risk'\n", " end AS RISK\n", " from {cust_data}\n", " ) cd \n", " left join\n", " (\n", " select customer_number as CUSTOMER_NUMBER_4, \n", " sar_flag as SAR_FLAG\n", " from {alert_data}\n", " ) ad \n", " on cd.CUSTOMER_NUMBER_3 = ad.CUSTOMER_NUMBER_4\n", " ) as cust_alert\n", " on cust_alert.CUSTOMER_NUMBER_cust = main.CUSTOMER_NUMBER_main\n", ") as final\n", "\"\"\"\n", "\n", "from tms_data_interface import SQLQueryInterface\n", "\n", "class Scenario:\n", " seq = SQLQueryInterface(schema=\"transactionschema\")\n", "\n", " def logic(self, **kwargs):\n", " row_list = self.seq.execute_raw(query.format(trans_data=\"transaction10m\",\n", " cust_data=\"customer_data_v1\",\n", " acc_data=\"account_data_v1\",\n", " alert_data=\"alert_data_v1\")\n", " )\n", " cols = [\"Focal_id\", \"Cash_deposit_total\", \"Cash_deposit_count\",\n", " \"Segment\", \"Risk\", \"SAR_FLAG\"]\n", " df = pd.DataFrame(row_list, columns = cols)\n", " df[\"Cash_deposit_total\"] = df[\"Cash_deposit_total\"].astype(float)\n", " return df" ] }, { "cell_type": "code", "execution_count": 3, "id": "1f20337b-8116-47e5-8743-1ba41e2df819", "metadata": { "tags": [] }, "outputs": [ { "data": { "text/html": [ "
| \n", " | Focal_id | \n", "Cash_deposit_total | \n", "Cash_deposit_count | \n", "Segment | \n", "Risk | \n", "SAR_FLAG | \n", "
|---|---|---|---|---|---|---|
| 0 | \n", "PN469304 | \n", "8049952.0 | \n", "141 | \n", "Mass Market | \n", "High Risk | \n", "N | \n", "
| 1 | \n", "PN605743 | \n", "9766101.0 | \n", "175 | \n", "Others | \n", "Low Risk | \n", "N | \n", "
| 2 | \n", "PN399215 | \n", "8469943.0 | \n", "164 | \n", "Mass Market | \n", "Low Risk | \n", "N | \n", "
| 3 | \n", "PN175020 | \n", "17724959.0 | \n", "313 | \n", "Others | \n", "Low Risk | \n", "N | \n", "
| 4 | \n", "PN345085 | \n", "7518628.0 | \n", "124 | \n", "Others | \n", "Medium Risk | \n", "N | \n", "
| ... | \n", "... | \n", "... | \n", "... | \n", "... | \n", "... | \n", "... | \n", "
| 9005 | \n", "PN660329 | \n", "8528178.0 | \n", "141 | \n", "Others | \n", "Low Risk | \n", "N | \n", "
| 9006 | \n", "PN336833 | \n", "8475036.0 | \n", "150 | \n", "Others | \n", "Low Risk | \n", "N | \n", "
| 9007 | \n", "PN428534 | \n", "7871170.0 | \n", "129 | \n", "Priority Banking | \n", "Low Risk | \n", "N | \n", "
| 9008 | \n", "PN101660 | \n", "8450309.0 | \n", "142 | \n", "Mass Market | \n", "Low Risk | \n", "N | \n", "
| 9009 | \n", "PN198641 | \n", "8690226.0 | \n", "166 | \n", "Others | \n", "Low Risk | \n", "N | \n", "
9010 rows × 6 columns
\n", "