Skip to content

Commit 9d11876

Browse files
committed
Rename fields for clarity
1 parent 2cf4634 commit 9d11876

File tree

2 files changed

+28
-13
lines changed

2 files changed

+28
-13
lines changed

nomenklatura/resolver/edge.py

+24-9
Original file line numberDiff line numberDiff line change
@@ -6,8 +6,16 @@
66

77

88
class Edge(object):
9-
10-
__slots__ = ("key", "source", "target", "judgement", "score", "user", "timestamp")
9+
__slots__ = (
10+
"key",
11+
"source",
12+
"target",
13+
"judgement",
14+
"score",
15+
"user",
16+
"created_at",
17+
"deleted_at",
18+
)
1119

1220
def __init__(
1321
self,
@@ -16,14 +24,16 @@ def __init__(
1624
judgement: Judgement = Judgement.NO_JUDGEMENT,
1725
score: Optional[float] = None,
1826
user: Optional[str] = None,
19-
timestamp: Optional[str] = None,
27+
created_at: Optional[str] = None,
28+
deleted_at: Optional[str] = None,
2029
):
2130
self.key = Identifier.pair(left_id, right_id)
2231
self.target, self.source = self.key
2332
self.judgement = judgement
2433
self.score = score
2534
self.user = user
26-
self.timestamp = timestamp
35+
self.created_at = created_at
36+
self.deleted_at = deleted_at
2737

2838
def other(self, cur: Identifier) -> Identifier:
2939
if cur == self.target:
@@ -37,7 +47,8 @@ def to_dict(self) -> Dict[str, Any]:
3747
"judgement": self.judgement.value,
3848
"score": self.score,
3949
"user": self.user,
40-
"timestamp": self.timestamp,
50+
"created_at": self.created_at,
51+
"deleted_at": self.deleted_at,
4152
}
4253

4354
def to_line(self) -> str:
@@ -47,7 +58,7 @@ def to_line(self) -> str:
4758
self.judgement.value,
4859
self.score,
4960
self.user,
50-
self.timestamp,
61+
self.created_at,
5162
]
5263
return json.dumps(row) + "\n"
5364

@@ -69,14 +80,17 @@ def __repr__(self) -> str:
6980
@classmethod
7081
def from_line(cls, line: str) -> "Edge":
7182
data = json.loads(line)
72-
return cls(
83+
edge = cls(
7384
data[0],
7485
data[1],
7586
judgement=Judgement(data[2]),
7687
score=data[3],
7788
user=data[4],
78-
timestamp=data[5],
89+
created_at=data[5],
7990
)
91+
if len(data) > 6:
92+
edge.deleted_at = data[6]
93+
return edge
8094

8195
@classmethod
8296
def from_dict(cls, data: Dict[str, Any]) -> "Edge":
@@ -86,5 +100,6 @@ def from_dict(cls, data: Dict[str, Any]) -> "Edge":
86100
judgement=Judgement(data["judgement"]),
87101
score=data["score"],
88102
user=data["user"],
89-
timestamp=data["timestamp"],
103+
created_at=data.get("created_at"),
104+
deleted_at=data.get("deleted_at"),
90105
)

nomenklatura/resolver/resolver.py

+4-4
Original file line numberDiff line numberDiff line change
@@ -54,7 +54,7 @@ def __init__(
5454
Column("judgement", Unicode(14), nullable=False),
5555
Column("score", Float, nullable=True),
5656
Column("user", Unicode(512), nullable=False),
57-
Column("timestamp", Unicode(28)),
57+
Column("created_at", Unicode(28)),
5858
Column("deleted_at", Unicode(28), nullable=True),
5959
extend_existing=True,
6060
)
@@ -359,7 +359,7 @@ def get_judgements(
359359
stmt = self._table.select()
360360
stmt = stmt.where(self._table.c.judgement != Judgement.NO_JUDGEMENT.value)
361361
stmt = stmt.where(self._table.c.deleted_at.is_(None))
362-
stmt = stmt.order_by(self._table.c.timestamp.desc())
362+
stmt = stmt.order_by(self._table.c.created_at.desc())
363363
if limit is not None:
364364
stmt = stmt.limit(limit)
365365
cursor = self._get_connection().execute(stmt)
@@ -440,7 +440,7 @@ def decide(
440440
return canonical
441441

442442
edge.judgement = judgement
443-
edge.timestamp = timestamp()
443+
edge.created_at = timestamp()
444444
edge.user = user or getpass.getuser()
445445
edge.score = score or edge.score
446446
self._register(edge)
@@ -456,7 +456,7 @@ def _register(self, edge: Edge) -> None:
456456
judgement=istmt.excluded.judgement,
457457
score=istmt.excluded.score,
458458
user=istmt.excluded.user,
459-
timestamp=istmt.excluded.timestamp,
459+
created_at=istmt.excluded.created_at,
460460
deleted_at=None,
461461
)
462462
stmt = istmt.on_conflict_do_update(

0 commit comments

Comments
 (0)