个人客户源行数

import json
import os
import sqlite3
from pathlib import Path

import pandas as pd

def _root():
    env = os.environ.get("ETL_FIXTURE_DIR")
    if env:
        return Path(env)
    return Path("/static/data/etl/bank-east-1104")

def _load_csv(name):
    p = _root() / name
    if p.is_file():
        return pd.read_csv(p, dtype=str).fillna("")
    return pd.read_csv(f"/static/data/etl/bank-east-1104/{name}", dtype=str).fillna("")

def _load_json(name):
    p = _root() / name
    if p.is_file():
        return json.loads(p.read_text(encoding="utf-8"))
    return pd.read_json(f"/static/data/etl/bank-east-1104/{name}").to_dict(orient="records")

def _db():
    con = sqlite3.connect(":memory:")
    con.executescript("""
    CREATE TABLE stg_org(nbjgh TEXT, yhjgmc TEXT, jrxkzh TEXT);
    CREATE TABLE stg_cif_indiv(cust_no TEXT, name TEXT, id_type TEXT, id_no TEXT, update_dt TEXT);
    CREATE TABLE stg_cif_corp(cust_no TEXT, name TEXT, uscc TEXT, update_dt TEXT);
    CREATE TABLE stg_deposit_acct(zh TEXT, cust_no TEXT, cust_type TEXT, nbjgh TEXT, zhye REAL, bz TEXT, biz_date TEXT, status TEXT);
    CREATE TABLE stg_deposit_txn(zh TEXT, jyrq TEXT, jyje REAL, jyfx TEXT, jydsxm TEXT, jydszh TEXT);
    CREATE TABLE stg_loan_contract(htbh TEXT, cust_no TEXT, cust_type TEXT, htje REAL, qsrq TEXT, dqrq TEXT, biz_date TEXT);
    CREATE TABLE stg_loan_acct(dkfhzh TEXT, htbh TEXT, cust_no TEXT, cust_type TEXT, dkye REAL, wjfl TEXT, biz_date TEXT);
    CREATE TABLE stg_gl_balance(km TEXT, kmmc TEXT, ye REAL, bgq TEXT);
    CREATE TABLE JGXXB(nbjgh TEXT, yhjgmc TEXT, jrxkzh TEXT, bgq TEXT);
    CREATE TABLE GRJCXXB(khtybh TEXT, khxm TEXT, zjlb TEXT, zjhm TEXT, bgq TEXT);
    CREATE TABLE DGKHXXB(khtybh TEXT, khmc TEXT, tyshxydm TEXT, bgq TEXT);
    CREATE TABLE GRCKFHZ(zh TEXT, khtybh TEXT, nbjgh TEXT, zhye REAL, bz TEXT, bgq TEXT);
    CREATE TABLE DGCKFHZ(zh TEXT, khtybh TEXT, nbjgh TEXT, zhye REAL, bz TEXT, bgq TEXT);
    CREATE TABLE GRCKFHZMX(zh TEXT, jyrq TEXT, jyje REAL, jyfx TEXT, jydsxm TEXT, jydszh TEXT);
    CREATE TABLE DGCKFHZMX(zh TEXT, jyrq TEXT, jyje REAL, jyfx TEXT, jydsxm TEXT, jydszh TEXT);
    CREATE TABLE XDHTB(htbh TEXT, khtybh TEXT, htje REAL, qsrq TEXT, dqrq TEXT, bgq TEXT);
    CREATE TABLE GRXDFHZ(dkfhzh TEXT, htbh TEXT, khtybh TEXT, dkye REAL, wjfl TEXT, bgq TEXT);
    CREATE TABLE DGXDFHZ(dkfhzh TEXT, htbh TEXT, khtybh TEXT, dkye REAL, wjfl TEXT, bgq TEXT);
    CREATE TABLE rpt_g01(xmdm TEXT, xmmc TEXT, ye REAL, bgq TEXT);
    CREATE TABLE rpt_loan_5class(wjfl TEXT, ye REAL, bgq TEXT);
    CREATE TABLE rpt_tieout(rule TEXT, east REAL, g01 REAL, gl REAL, diff REAL, pass INTEGER, bgq TEXT);
    CREATE TABLE dq_reject(rule TEXT, tbl TEXT, pk TEXT, reason TEXT, bgq TEXT);
    """)
    return con

def solve():
    df = _load_csv('cif_indiv.csv')
    return len(df)