o
    Ղi                     @   s<   d Z ddlmZ ddlZdZdZdZdZdd Z	dd	 Z
dS )
zgadd data_sources table and extend tax_documents

Revision ID: 006
Revises: 005
Create Date: 2026-04-20
    )opN006005c                   C   s  t jdtjdt ddtjdtddddtjd	td
ddtjdtdddtjdtdddtjdtdddtjdtd
ddtjdt ddtjdt dt dtjdtdddtjdt	 dddtjdt	 dddtjdt dddtjdt dt
 dtjdt ddtjdt ddtddd t d  t jd!tjd"t td#dddd t jd!tjd$tddddd t jd!tjdtddddd t jd!tjd%t dddd t jd!tjd&t dddd t d' t jd(d!d"gdd t jd)d!d$gdd t jd*d!dgdd t jd+tjd"t td#dddd d S ),Ndata_sourcesidF)nullablecode2   T)r   uniquename   source_typeregion_level   region_codeadapter_classadapter_confighas_categories)r   server_defaultcrawl_scheduled   request_delay_minz1.0request_delay_maxz5.0max_retries3	is_activelast_crawled_at
created_atdata_centerschemau  INSERT INTO data_center.data_sources (id, code, name, source_type, region_level, region_code, adapter_class, has_categories, is_active) VALUES (1, 'chinatax', '国家税务总局', 'regulation', 'national', 'CN', 'adapters.chinatax.adapter.ChinataxAdapter', TRUE, TRUE)tax_documents	source_idzdata_center.data_sources.iddoc_typeqa_question	qa_answerzqUPDATE data_center.tax_documents SET source_id=1, doc_type='regulation', region_code='CN' WHERE source_id IS NULLidx_doc_source_ididx_doc_doc_typeidx_doc_region_codedata_processing_tasks)r   create_tablesaColumnIntegerStringJSONBooleanfalseFloattrueDateTimePrimaryKeyConstraintexecute
add_column
ForeignKeyTextcreate_index r;   r;   c/lsinfo/ai/hellotax_ai/data_center/backend/alembic/versions/006_add_datasource_and_extend_taxdoc.pyupgrade   sH   (""  ,r=   c                   C   s   t jdddd t jdddd t jdddd t jd	ddd t jdd
dd t jdddd t jdddd t jdddd t jdddd t jddd d S )Nr)   r"   r   r   r(   r!   )
table_namer    r'   r&   r%   r$   r   r#   r   )r   drop_column
drop_index
drop_tabler;   r;   r;   r<   	downgradeF   s   rB   )__doc__alembicr   
sqlalchemyr+   revisiondown_revisionbranch_labels
depends_onr=   rB   r;   r;   r;   r<   <module>   s    6