test_capacity.py 829 B

12345678910111213141516171819
  1. from BaseDataMaintenance.common.Utils import getRow_ots
  2. from BaseDataMaintenance.dataSource.source import getConnect_ots_capacity,getConnect_ots
  3. from tablestore import *
  4. ots_client = getConnect_ots()
  5. query = BoolQuery(must_queries=[TermQuery("page_time","2022-12-02")],
  6. should_queries=[TermQuery("province","广西"),
  7. TermQuery("province","广东")])
  8. rows,next_token,total_count,is_all_succeed = ots_client.search("project2","project2_index",
  9. SearchQuery(query,get_total_count=True,limit=100),
  10. columns_to_get=ColumnsToGet(column_names=["page_time","province"],return_type=ColumnReturnType.SPECIFIED))
  11. print(total_count)
  12. print(getRow_ots(rows))