修改acturetime/data 入库无值问题
This commit is contained in:
@@ -1,6 +1,7 @@
|
||||
import time
|
||||
import time
|
||||
from dataclasses import dataclass
|
||||
from datetime import datetime
|
||||
from decimal import Decimal, InvalidOperation
|
||||
|
||||
|
||||
DATA_KEYS = [
|
||||
@@ -52,6 +53,17 @@ DATA_KEYS = [
|
||||
"space5",
|
||||
]
|
||||
|
||||
DATA_KEY_ALIASES = {
|
||||
"actual_date": ("actual_date", "actualDate"),
|
||||
"actual_time": ("actual_time", "actualTime"),
|
||||
"wellid": ("wellid", "wellId"),
|
||||
}
|
||||
|
||||
META_KEY_ALIASES = {
|
||||
"equipment_code": ("equipment_code", "equipmentCode"),
|
||||
"equipment_sn": ("equipment_sn", "equipmentSN"),
|
||||
}
|
||||
|
||||
|
||||
@dataclass(frozen=True)
|
||||
class DrillingRealtimeData:
|
||||
@@ -60,8 +72,8 @@ class DrillingRealtimeData:
|
||||
stknum: int
|
||||
recid: int
|
||||
seqid: int
|
||||
actual_date: float
|
||||
actual_time: float
|
||||
actual_date: int
|
||||
actual_time: int
|
||||
actcod: int
|
||||
deptbitm: float
|
||||
deptbitv: float
|
||||
@@ -111,8 +123,8 @@ class DrillingRealtimeData:
|
||||
stknum=0,
|
||||
recid=0,
|
||||
seqid=0,
|
||||
actual_date=float(now.strftime("%Y%m%d")),
|
||||
actual_time=float(now.strftime("%H%M%S")),
|
||||
actual_date=int(now.strftime("%Y%m%d")),
|
||||
actual_time=int(now.strftime("%H%M%S")),
|
||||
actcod=0,
|
||||
deptbitm=0.0,
|
||||
deptbitv=0.0,
|
||||
@@ -156,11 +168,21 @@ class DrillingRealtimeData:
|
||||
|
||||
@classmethod
|
||||
def from_payload(cls, payload):
|
||||
meta = payload.get("meta") if isinstance(payload, dict) and isinstance(payload.get("meta"), dict) else {}
|
||||
meta = get_dict_alias(payload, "meta", "mate")
|
||||
data = payload.get("data") if isinstance(payload, dict) and isinstance(payload.get("data"), dict) else {}
|
||||
values = {key: data.get(key, 0) for key in DATA_KEYS}
|
||||
values["ts"] = int(data.get("ts", data.get("record_time", int(time.time() * 1000))))
|
||||
values["wellid"] = data.get("wellid") or meta.get("equipment_code") or meta.get("equipment_sn") or ""
|
||||
values = {key: get_alias_value(data, DATA_KEY_ALIASES.get(key, (key,)), 0) for key in DATA_KEYS}
|
||||
values["ts"] = parse_int_value(
|
||||
get_alias_value(data, ("ts", "record_time", "recordTime"), int(time.time() * 1000)),
|
||||
int(time.time() * 1000),
|
||||
)
|
||||
values["actual_date"] = parse_int_value(get_alias_value(data, DATA_KEY_ALIASES["actual_date"], 0))
|
||||
values["actual_time"] = parse_int_value(get_alias_value(data, DATA_KEY_ALIASES["actual_time"], 0))
|
||||
values["wellid"] = (
|
||||
get_alias_value(data, DATA_KEY_ALIASES["wellid"], "")
|
||||
or get_alias_value(meta, META_KEY_ALIASES["equipment_code"], "")
|
||||
or get_alias_value(meta, META_KEY_ALIASES["equipment_sn"], "")
|
||||
or ""
|
||||
)
|
||||
return cls(**values)
|
||||
|
||||
def to_payload(self, equipment_code):
|
||||
@@ -174,3 +196,27 @@ class DrillingRealtimeData:
|
||||
for key in DATA_KEYS
|
||||
},
|
||||
}
|
||||
|
||||
|
||||
def parse_int_value(value, default=0):
|
||||
try:
|
||||
if value is None or value == "":
|
||||
return default
|
||||
return int(Decimal(str(value).strip()))
|
||||
except (InvalidOperation, TypeError, ValueError):
|
||||
return default
|
||||
|
||||
|
||||
def get_alias_value(source, keys, default=None):
|
||||
if not isinstance(source, dict):
|
||||
return default
|
||||
for key in keys:
|
||||
value = source.get(key)
|
||||
if value is not None:
|
||||
return value
|
||||
return default
|
||||
|
||||
|
||||
def get_dict_alias(source, *keys):
|
||||
value = get_alias_value(source, keys, {})
|
||||
return value if isinstance(value, dict) else {}
|
||||
|
||||
@@ -43,8 +43,8 @@ class WitsData:
|
||||
stknum: int
|
||||
recid: int
|
||||
seqid: int
|
||||
actual_date: str
|
||||
actual_time: str
|
||||
actual_date: int
|
||||
actual_time: int
|
||||
actual_ts: int
|
||||
actcod: int
|
||||
actod_label: str
|
||||
@@ -122,8 +122,8 @@ WITS_CHANNEL_MAPPING = [
|
||||
("0102", "stknum", "int"),
|
||||
("0103", "recid", "int"),
|
||||
("0104", "seqid", "int"),
|
||||
("0105", "actual_date", "string"),
|
||||
("0106", "actual_time", "string"),
|
||||
("0105", "actual_date", "int"),
|
||||
("0106", "actual_time", "int"),
|
||||
("0107", "actcod", "int"),
|
||||
("0108", "deptbitm", "float6"),
|
||||
("0109", "deptbitv", "float6"),
|
||||
|
||||
Reference in New Issue
Block a user