Updata2

tmp/add_missing_ods_columns.py

@@ -0,0 +1,162 @@
+# -*- coding: utf-8 -*-
+"""
+添加缺失的 ODS 列到数据库
+"""
+import psycopg2
+
+DSN = 'postgresql://local-Python:Neo-local-1991125@100.64.0.4:5432/LLZQ-test'
+
+# 缺失字段定义：表名 -> [(列名, 类型, 注释)]
+MISSING_COLUMNS = {
+    'billiards_ods.settlement_records': [
+        ('electricityadjustmoney', 'NUMERIC(18,2)', '电费调整金额'),
+        ('electricitymoney', 'NUMERIC(18,2)', '电费金额'),
+        ('mervousalesamount', 'NUMERIC(18,2)', '商户券销售额'),
+        ('plcouponsaleamount', 'NUMERIC(18,2)', '平台券销售额'),
+        ('realelectricitymoney', 'NUMERIC(18,2)', '实际电费金额'),
+        ('settlelist', 'JSONB', '结算明细列表'),
+    ],
+    'billiards_ods.recharge_settlements': [
+        ('electricityadjustmoney', 'NUMERIC(18,2)', '电费调整金额'),
+        ('electricitymoney', 'NUMERIC(18,2)', '电费金额'),
+        ('mervousalesamount', 'NUMERIC(18,2)', '商户券销售额'),
+        ('plcouponsaleamount', 'NUMERIC(18,2)', '平台券销售额'),
+        ('realelectricitymoney', 'NUMERIC(18,2)', '实际电费金额'),
+        ('settlelist', 'JSONB', '结算明细列表'),
+    ],
+    'billiards_ods.table_fee_transactions': [
+        ('activity_discount_amount', 'NUMERIC(18,2)', '活动折扣金额'),
+        ('order_consumption_type', 'INT', '订单消费类型'),
+        ('real_service_money', 'NUMERIC(18,2)', '实际服务费金额'),
+    ],
+    'billiards_ods.assistant_service_records': [
+        ('assistantteamname', 'TEXT', '助教团队名称'),
+        ('real_service_money', 'NUMERIC(18,2)', '实际服务费金额'),
+    ],
+    'billiards_ods.group_buy_redemption_records': [
+        ('assistant_service_share_money', 'NUMERIC(18,2)', '助教服务分摊金额'),
+        ('assistant_share_money', 'NUMERIC(18,2)', '助教分摊金额'),
+        ('coupon_sale_id', 'BIGINT', '优惠券销售ID'),
+        ('good_service_share_money', 'NUMERIC(18,2)', '商品服务分摊金额'),
+        ('goods_share_money', 'NUMERIC(18,2)', '商品分摊金额'),
+        ('member_discount_money', 'NUMERIC(18,2)', '会员折扣金额'),
+        ('recharge_share_money', 'NUMERIC(18,2)', '充值分摊金额'),
+        ('table_service_share_money', 'NUMERIC(18,2)', '台费服务分摊金额'),
+        ('table_share_money', 'NUMERIC(18,2)', '台费分摊金额'),
+    ],
+    'billiards_ods.table_fee_discount_records': [
+        ('area_type_id', 'BIGINT', '区域类型ID'),
+        ('charge_free', 'BOOLEAN', '是否免费'),
+        ('site_table_area_id', 'BIGINT', '门店台区ID'),
+        ('site_table_area_name', 'TEXT', '门店台区名称'),
+        ('sitename', 'TEXT', '门店名称'),
+        ('table_name', 'TEXT', '台桌名称'),
+        ('table_price', 'NUMERIC(18,2)', '台桌价格'),
+        ('tenant_name', 'TEXT', '租户名称'),
+    ],
+    'billiards_ods.member_stored_value_cards': [
+        ('able_share_member_discount', 'BOOLEAN', '是否可共享会员折扣'),
+        ('electricity_deduct_radio', 'NUMERIC(18,4)', '电费扣减比例'),
+        ('electricity_discount', 'NUMERIC(18,4)', '电费折扣'),
+        ('electricitycarddeduct', 'BOOLEAN', '电费卡扣'),
+        ('member_grade', 'INT', '会员等级'),
+        ('principal_balance', 'NUMERIC(18,2)', '本金余额'),
+        ('rechargefreezebalance', 'NUMERIC(18,2)', '充值冻结余额'),
+    ],
+    'billiards_ods.member_profiles': [
+        ('pay_money_sum', 'NUMERIC(18,2)', '累计支付金额'),
+        ('person_tenant_org_id', 'BIGINT', '人员租户组织ID'),
+        ('person_tenant_org_name', 'TEXT', '人员租户组织名称'),
+        ('recharge_money_sum', 'NUMERIC(18,2)', '累计充值金额'),
+        ('register_source', 'TEXT', '注册来源'),
+    ],
+    'billiards_ods.member_balance_changes': [
+        ('principal_after', 'NUMERIC(18,2)', '变动后本金'),
+        ('principal_before', 'NUMERIC(18,2)', '变动前本金'),
+        ('principal_data', 'TEXT', '本金变动数据'),
+    ],
+    'billiards_ods.group_buy_packages': [
+        ('is_first_limit', 'BOOLEAN', '是否首单限制'),
+        ('sort', 'INT', '排序'),
+        ('tenantcouponsaleorderitemid', 'BIGINT', '租户券销售订单项ID'),
+    ],
+    'billiards_ods.store_goods_master': [
+        ('commodity_code', 'TEXT', '商品编码'),
+        ('not_sale', 'BOOLEAN', '是否停售'),
+    ],
+    'billiards_ods.assistant_cancellation_records': [
+        ('tenant_id', 'BIGINT', '租户ID'),
+    ],
+    'billiards_ods.store_goods_sales_records': [
+        ('coupon_share_money', 'NUMERIC(18,2)', '优惠券分摊金额'),
+    ],
+    'billiards_ods.payment_transactions': [
+        ('tenant_id', 'BIGINT', '租户ID'),
+    ],
+    'billiards_ods.site_tables_master': [
+        ('order_id', 'BIGINT', '订单ID'),
+    ],
+    'billiards_ods.tenant_goods_master': [
+        ('not_sale', 'BOOLEAN', '是否停售'),
+    ],
+}
+
+
+def get_existing_columns(conn, schema, table):
+    """获取表已有的列"""
+    sql = """
+        SELECT column_name
+        FROM information_schema.columns
+        WHERE table_schema = %s AND table_name = %s
+    """
+    with conn.cursor() as cur:
+        cur.execute(sql, (schema, table))
+        return {row[0].lower() for row in cur.fetchall()}
+
+
+def add_column(conn, full_table, col_name, col_type, comment):
+    """添加列"""
+    schema, table = full_table.split('.')
+    sql = f'ALTER TABLE {full_table} ADD COLUMN IF NOT EXISTS "{col_name}" {col_type}'
+    comment_sql = f"COMMENT ON COLUMN {full_table}.\"{col_name}\" IS '{comment}'"
+    
+    with conn.cursor() as cur:
+        cur.execute(sql)
+        cur.execute(comment_sql)
+    conn.commit()
+    print(f"  [OK] 添加列: {col_name} ({col_type})")
+
+
+def main():
+    conn = psycopg2.connect(DSN)
+    
+    print("=" * 80)
+    print("添加缺失的 ODS 列")
+    print("=" * 80)
+    
+    total_added = 0
+    total_skipped = 0
+    
+    for full_table, columns in MISSING_COLUMNS.items():
+        schema, table = full_table.split('.')
+        print(f"\n处理表: {full_table}")
+        
+        existing = get_existing_columns(conn, schema, table)
+        
+        for col_name, col_type, comment in columns:
+            if col_name.lower() in existing:
+                print(f"  [跳过] 列已存在: {col_name}")
+                total_skipped += 1
+            else:
+                add_column(conn, full_table, col_name, col_type, comment)
+                total_added += 1
+    
+    conn.close()
+    
+    print("\n" + "=" * 80)
+    print(f"完成: 添加 {total_added} 列, 跳过 {total_skipped} 列")
+    print("=" * 80)
+
+
+if __name__ == '__main__':
+    main()