add more logs
This commit is contained in:
@@ -5,6 +5,7 @@ import firebase_admin
|
||||
from firebase_admin import credentials, firestore
|
||||
|
||||
import definitions
|
||||
import params
|
||||
from pojo import ResultEnum
|
||||
from pojo.MailPojo import MailPojo
|
||||
from pojo.ReserveResultPojo import ReserveResultPojo
|
||||
@@ -27,25 +28,30 @@ class DataManager:
|
||||
self._db = firestore.client()
|
||||
|
||||
def get_all_sim_infos(self):
|
||||
params.oracle_log_sender.send_read_db_event("get_all_sim_infos")
|
||||
sim_info_collection = self._db.collection(SIM_INFOS)
|
||||
return sim_info_collection
|
||||
|
||||
def get_all_successful_items(self):
|
||||
params.oracle_log_sender.send_read_db_event("get_all_successful_items")
|
||||
return self.get_all_successful_items_for_day(str(datetime.date.today()), None)
|
||||
|
||||
def get_all_successful_items_for_day(self, day, source_from: Union[str, None]):
|
||||
params.oracle_log_sender.send_read_db_event("get_all_successful_items_for_day for {}".format(day))
|
||||
doc_ref = self._db.collection(day)
|
||||
if source_from is not None:
|
||||
doc_ref.where(u'source_from', u'==', source_from)
|
||||
return doc_ref
|
||||
|
||||
def get_successful_item_for_day_by_status(self, day, status: ResultEnum):
|
||||
params.oracle_log_sender.send_read_db_event("get_successful_item_for_day_by_status for {}".format(day))
|
||||
doc_ref = self._db.collection(day)
|
||||
if status is not None:
|
||||
doc_ref.where(u'accepted', u'==', status.value)
|
||||
return doc_ref
|
||||
|
||||
def save_sim_info(self, sim_info: SimInfoPojo):
|
||||
params.oracle_log_sender.send_read_db_event("save_sim_info")
|
||||
doc_ref = self._db.collection(SIM_INFOS).document(sim_info.phone)
|
||||
doc_ref.set(sim_info.to_firestore_dict())
|
||||
|
||||
@@ -57,6 +63,7 @@ class DataManager:
|
||||
doc_ref.set(result.to_firestore_dict())
|
||||
|
||||
def find_appointment_detail_via_phone(self, day, phone) -> ReserveResultPojo:
|
||||
params.oracle_log_sender.send_read_db_event("find_appointment_detail_via_phone")
|
||||
doc_ref = self._db.collection(day)
|
||||
results = doc_ref.where(u'phone', u'==', phone).stream()
|
||||
result_list = []
|
||||
@@ -70,6 +77,7 @@ class DataManager:
|
||||
doc_ref.set(contact.to_firestore_dict())
|
||||
|
||||
def _delete_collection(self, coll_ref, batch_size):
|
||||
params.oracle_log_sender.send_read_db_event("_delete_collection")
|
||||
docs = coll_ref.limit(batch_size).stream()
|
||||
deleted = 0
|
||||
|
||||
@@ -83,14 +91,17 @@ class DataManager:
|
||||
|
||||
# 删除数据库中所有的sim卡信息
|
||||
def clear_all_sim_info(self):
|
||||
params.oracle_log_sender.send_read_db_event("clear_all_sim_info")
|
||||
coll_ref = self._db.collection(SIM_INFOS)
|
||||
self._delete_collection(coll_ref, self.batch_size)
|
||||
|
||||
def read_contacts_from_db(self) -> list:
|
||||
params.oracle_log_sender.send_read_db_event("read_contacts_from_db")
|
||||
contact_collection = self._db.collection(CONTACT_COLLECTION_NAME)
|
||||
return contact_collection
|
||||
|
||||
def get_mail_list(self) -> list:
|
||||
params.oracle_log_sender.send_read_db_event("get_mail_list")
|
||||
mail_collection = self._db.collection(MAIL_COLLECTION_NAME)
|
||||
mail_list = []
|
||||
for mail in mail_collection.stream():
|
||||
|
||||
Reference in New Issue
Block a user