You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
57 lines
2.0 KiB
57 lines
2.0 KiB
import random
|
|
from fe.access import book
|
|
from fe.access.new_seller import register_new_seller
|
|
|
|
|
|
class GenBook:
|
|
def __init__(self, user_id, store_id):
|
|
self.user_id = user_id
|
|
self.store_id = store_id
|
|
self.password = self.user_id
|
|
self.seller = register_new_seller(self.user_id, self.password)
|
|
code = self.seller.create_store(store_id)
|
|
assert code == 200
|
|
self.__init_book_list__()
|
|
|
|
def __init_book_list__(self):
|
|
self.buy_book_info_list = []
|
|
self.buy_book_id_list = []
|
|
|
|
def gen(self, non_exist_book_id: bool, low_stock_level, max_book_count: int = 100) -> (bool, []):
|
|
self.__init_book_list__()
|
|
ok = True
|
|
book_db = book.BookDB()
|
|
rows = book_db.get_book_count()
|
|
start = 0
|
|
if rows > max_book_count:
|
|
start = random.randint(0, rows - max_book_count)
|
|
size = random.randint(1, max_book_count)
|
|
books = book_db.get_book_info(start, size)
|
|
book_id_exist = []
|
|
book_id_stock_level = {}
|
|
for bk in books:
|
|
if low_stock_level:
|
|
stock_level = random.randint(0, 100)
|
|
else:
|
|
stock_level = random.randint(2, 100)
|
|
code = self.seller.add_book(self.store_id, stock_level, bk)
|
|
assert code == 200
|
|
book_id_stock_level[bk.id] = stock_level
|
|
book_id_exist.append(bk)
|
|
|
|
for bk in book_id_exist:
|
|
stock_level = book_id_stock_level[bk.id]
|
|
if stock_level > 1:
|
|
buy_num = random.randint(1, stock_level)
|
|
else:
|
|
buy_num = 0
|
|
# add a new pair
|
|
if non_exist_book_id:
|
|
bk.id = bk.id + "_x"
|
|
if low_stock_level:
|
|
buy_num = stock_level + 1
|
|
self.buy_book_info_list.append((bk, buy_num))
|
|
|
|
for item in self.buy_book_info_list:
|
|
self.buy_book_id_list.append((item[0].id, item[1]))
|
|
return ok, self.buy_book_id_list
|