commspt-scripts/oci_email_statistics.py at main · LittleSkinCommspt/commspt-scripts · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
from datetime import datetime, timedelta
from zoneinfo import ZoneInfo

from oci.config import from_file
from oci.loggingsearch import LogSearchClient
from oci.loggingsearch.models import SearchLogsDetails

from utils.config_manager import config
from utils.send_message import group_message

oci_config = from_file("./.oci/config")

client = LogSearchClient(oci_config)

compartment_id = oci_config["tenancy"]

# time range
timezone = ZoneInfo("Asia/Shanghai")
datetime_end = datetime.now(timezone).replace(hour=5, minute=0, second=0, microsecond=0)  # Today 05:00:00
datetime_start = datetime_end - timedelta(days=1)  # Yesterday 05:00:00

log_stream_accepted = f"{compartment_id}/{config.oci.log_group_id}/{config.oci.log_object_id_accepted}"
log_stream_relayed = f"{compartment_id}/{config.oci.log_group_id}/{config.oci.log_object_id_relayed}"

details_accepted_total = SearchLogsDetails(
    search_query=f'search "{log_stream_accepted}" | count', time_start=datetime_start, time_end=datetime_end
)
details_accepted_suppressed = SearchLogsDetails(
    search_query=f"search \"{log_stream_accepted}\" | where data.errorType='Recipient suppressed' | count",
    time_start=datetime_start,
    time_end=datetime_end,
)
details_accepted_recipients = SearchLogsDetails(
    search_query=f'search "{log_stream_accepted}" | summarize count() by data.recipient | count',
    time_start=datetime_start,
    time_end=datetime_end,
)

details_relayed_total = SearchLogsDetails(
    search_query=f'search "{log_stream_relayed}" | count', time_start=datetime_start, time_end=datetime_end
)
details_relayed_bounced = SearchLogsDetails(
    search_query=f"search \"{log_stream_relayed}\" | where data.action='bounce' | count",
    time_start=datetime_start,
    time_end=datetime_end,
)
details_relayed_recipients = SearchLogsDetails(
    search_query=f'search "{log_stream_relayed}" | summarize count() by data.recipient | count',
    time_start=datetime_start,
    time_end=datetime_end,
)


resp_accepted_total = client.search_logs(details_accepted_total)
resp_accepted_suppressed = client.search_logs(details_accepted_suppressed)
resp_accepted_recipients = client.search_logs(details_accepted_recipients)
resp_relayed_total = client.search_logs(details_relayed_total)
resp_relayed_bounced = client.search_logs(details_relayed_bounced)
resp_relayed_recipients = client.search_logs(details_relayed_recipients)

# assert
assert resp_accepted_total, "Failed to get accepted total"
assert resp_accepted_suppressed, "Failed to get accepted suppressed"
assert resp_accepted_recipients, "Failed to get accepted recipients"
assert resp_relayed_total, "Failed to get relayed total"
assert resp_relayed_bounced, "Failed to get relayed bounced"
assert resp_relayed_recipients, "Failed to get relayed recipients"

accepted_total = resp_accepted_total.data.results[0].data["count"]
accepted_suppressed = resp_accepted_suppressed.data.results[0].data["count"]
accepted_recipients = resp_accepted_recipients.data.results[0].data["count"]
relayed_total = resp_relayed_total.data.results[0].data["count"]
relayed_bounced = resp_relayed_bounced.data.results[0].data["count"]
relayed_recipients = resp_relayed_recipients.data.results[0].data["count"]

# send message
message = f"""📊 OCI Email 统计
--- {datetime_start.strftime("%m-%d %H:%M")} ~ {datetime_end.strftime("%m-%d %H:%M")} ---

Accepted Total: {accepted_total}
Accepted Suppressed: {accepted_suppressed}
Accepted Suppression Rate: {round(accepted_suppressed / accepted_total * 100, 2)}%
Accepted Recipients: {accepted_recipients}

Relayed Total: {relayed_total}
Relayed Bounced: {relayed_bounced}
Relayed Bounce Rate: {round(relayed_bounced / relayed_total * 100, 2)}%
Relayed Recipients: {relayed_recipients}
"""

print(message)

group_message(config.groups_ids.commspt, message)