Skip to content

Commit

Permalink
tests fixed
Browse files Browse the repository at this point in the history
function order changed
  • Loading branch information
liampauling committed Jan 5, 2021
1 parent 5fbbcbe commit 1eae702
Show file tree
Hide file tree
Showing 3 changed files with 148 additions and 114 deletions.
45 changes: 23 additions & 22 deletions betfairlightweight/streaming/cache.py
Original file line number Diff line number Diff line change
Expand Up @@ -36,28 +36,11 @@ def __init__(self, prices: list, deletion_select: int, reverse: bool = False):
self.serialised = []
self.update(prices or [])

def serialise(self) -> None:
# avoiding dots / create local vars
s_deletion_select = self.deletion_select
self.serialised = [
{
"price": price,
"size": volume[s_deletion_select],
}
for price, volume in self.order_book.items()
]

def clear(self) -> None:
self.order_book = {}
self.serialise()

def _sort_order_book(self):
self.order_book = dict(sorted(self.order_book.items(), reverse=self.reverse))

def update(self, book_update: list) -> None:
deletion_select = self.deletion_select # local vars
for book in book_update:
price = book[0]
if book[self.deletion_select] == 0:
price = book[deletion_select - 1]
if book[deletion_select] == 0:
# remove price/size
try:
del self.order_book[price]
Expand All @@ -74,6 +57,24 @@ def update(self, book_update: list) -> None:
self.order_book[price] = book
self.serialise()

def clear(self) -> None:
self.order_book = {}
self.serialise()

def serialise(self) -> None:
# avoiding dots / create local vars
s_deletion_select = self.deletion_select
self.serialised = [
{
"price": price,
"size": volume[s_deletion_select],
}
for price, volume in self.order_book.items()
]

def _sort_order_book(self):
self.order_book = dict(sorted(self.order_book.items(), reverse=self.reverse))


class RunnerBook:
def __init__(
Expand Down Expand Up @@ -467,8 +468,8 @@ def serialise_orders(self, market_id: str) -> list:
def serialise_matches(self) -> dict:
return {
"selectionId": self.selection_id,
"matchedLays": self.matched_lays.serialise,
"matchedBacks": self.matched_backs.serialise,
"matchedLays": self.matched_lays.serialised,
"matchedBacks": self.matched_backs.serialised,
}


Expand Down

0 comments on commit 1eae702

Please sign in to comment.