aljaz/python-teos
1
0
Fork 0
mirror of https://github.com/aljazceru/python-teos.git synced 2025-12-17 06:04:21 +01:00
Code Issues Packages Projects Releases Wiki Activity

Fixes slots difference calculation

Browse Source
This commit is contained in:
Sergi Delgado Segura
2020-03-30 17:56:47 +02:00
parent 93a389508e
commit c9b3bb625f
1 changed files with 9 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
teos/api.py
View File

@@ -1,6 +1,6 @@
import os
import logging
from math import ceil
from math import ceil, floor
from flask import Flask, request, abort, jsonify
import teos.errors as errors
@@ -175,8 +175,10 @@ class API:
# For updates we only reserve the slot difference provided the new one is bigger.
if appointment_summary:
size_diff = len(appointment.encrypted_blob.data) - appointment_summary.get("size")
slot_diff = ceil(size_diff / ENCRYPTED_BLOB_MAX_SIZE_HEX)
used_slots = ceil(appointment_summary.get("size") / ENCRYPTED_BLOB_MAX_SIZE_HEX)
required_slots = ceil(len(appointment.encrypted_blob.data) / ENCRYPTED_BLOB_MAX_SIZE_HEX)
slot_diff = required_slots - used_slots
required_slots = slot_diff if slot_diff > 0 else 0
# For regular appointments 1 slot is reserved per ENCRYPTED_BLOB_MAX_SIZE_HEX block.
@@ -191,6 +193,10 @@ class API:
appointment_added, signature = self.watcher.add_appointment(appointment, user_pk)
if appointment_added:
# If the appointment is added and the update is smaller than the original, the difference is given back.
if slot_diff < 0:
self.gatekeeper.free_slots(user_pk, abs(slot_diff))
rcode = HTTP_OK
response = {
"locator": appointment.locator,
@@ -198,10 +204,6 @@ class API:
"available_slots": self.gatekeeper.registered_users[user_pk],
}
# If the appointment is added and the update is smaller than the original, the difference is given back.
if slot_diff < 0:
self.gatekeeper.free_slots(slot_diff)
else:
# If the appointment is not added the reserved slots are given back
self.gatekeeper.free_slots(user_pk, required_slots)