def save_image(self, label):
if not self.preview:
pop_up(
"Mark an image first.",
"Before you can save an image, you need to take a screenshot (with the take screenshot button),"
"Then you need to mark the top left corner of the poker window (or load a previously saved one)"
"After that an image needs to be marked by clicking on the top left and then bottom right corner of that image."
)
return
log.info(f"flattening button {label}")
self.signal_flatten_button.emit(label)
self.table_name = self.ui.table_name.currentText()
mongo = MongoManager()
owner = mongo.get_table_owner(self.table_name)
if owner != COMPUTER_NAME:
pop_up(
"Not authorized.",
"You can only edit your own tables. Please create a new copy or start with a new blank table"
)
return
log.info(f"Saving {label}")
mongo.update_table_image(pil_image=self.preview,
label=label,
table_name=self.table_name)
log.info("Saving complete")
def test_card_upload(): # OK!
mongo = MongoManager()
table_dict = mongo.get_table('GG_TEST')
table_scraper = TableScraper(table_dict)
# table_scraper.screenshot = Image.open(os.path.join(get_dir('tests', 'screenshots'), 'test5.png'))
# screnshot = table_scraper.crop_from_top_left_corner()
CARD_VALUES = "23456789TJQKA"
CARD_SUITES = "CDHS"
for value in CARD_VALUES:
for suit in CARD_SUITES:
fileName = value.upper() + suit.lower() + '.png'
label = value.lower() + suit.lower()
pil_image = Image.open(os.path.join('../../../card-detector\\images', fileName))
mongo.update_table_image(pil_image, label=label, table_name='GG_TEST')
