hash of beatmap file
This commit is contained in:
parent
a94baeac62
commit
fddb66fb86
|
@ -7,11 +7,26 @@ from .helpers import osuMods
|
|||
from .helpers import osuRanks
|
||||
from .helpers import typeSerializer
|
||||
|
||||
from io import BufferedReader
|
||||
from os.path import isfile
|
||||
from hashlib import md5 as _md5
|
||||
|
||||
def md5(str):
|
||||
return _md5(str.encode("ascii")).hexdigest().encode()
|
||||
def md5_str(data):
|
||||
if type(data) is str:
|
||||
data = data.encode("ascii")
|
||||
return _md5(data).hexdigest().encode()
|
||||
|
||||
def md5_file(file):
|
||||
if type(file) is not BufferedReader:
|
||||
raise Exception("File is not a BufferedReader")
|
||||
|
||||
if file.mode != "rb":
|
||||
raise Exception("File is not in ReadBytes mode")
|
||||
|
||||
hash = _md5()
|
||||
for chunk in iter(lambda: file.read(4096), b""):
|
||||
hash.update(chunk)
|
||||
return hash.hexdigest()
|
||||
|
||||
class Replay:
|
||||
mode = osuModes.STANDARD # Byte
|
||||
|
@ -31,7 +46,7 @@ class Replay:
|
|||
perfect = True # Byte
|
||||
|
||||
mods = osuMods.NOMOD # Int
|
||||
|
||||
|
||||
lifebar_graph = b"0|1," # Byte[]
|
||||
timestamp = 0 # Long
|
||||
|
||||
|
@ -50,58 +65,64 @@ class Replay:
|
|||
for k, v in kwargs.items():
|
||||
if k in allowed_kwargs.keys():
|
||||
allowed_kwargs[k](v) # Run callback func
|
||||
|
||||
|
||||
# Set validators -----------------------------------------------------------
|
||||
|
||||
def set_mode(self, mode_id):
|
||||
if mode_id > 3 or mode_id < 0:
|
||||
raise Exception("Invalid mode")
|
||||
self.mode_id = mode_id
|
||||
|
||||
|
||||
def set_osu_version(self, osu_version):
|
||||
if type(osu_version) is not int:
|
||||
raise Exception("osu! version must be an int")
|
||||
self.osu_version = osu_version
|
||||
|
||||
|
||||
def set_beatmap_hash(self, md5_hash):
|
||||
if len(md5_hash) != 32 or len([x for x in md5_hash if x in string.hexdigits]) != 32:
|
||||
raise Exception("Invalid beatmap hash")
|
||||
self.beatmap_hash = md5_hash
|
||||
|
||||
|
||||
def set_beatmap(self, filepath):
|
||||
if not isfile(filepath):
|
||||
raise Exception("Beatmap file not found")
|
||||
with open(filepath, "rb") as f:
|
||||
self.beatmap_hash = md5_file(f)
|
||||
|
||||
def set_player_name(self, player_name):
|
||||
self.player_name = player_name
|
||||
|
||||
|
||||
def set_score_hash(self, md5_hash):
|
||||
if type(md5_hash) is bytes:
|
||||
md5_hash = md5_hash.decode()
|
||||
if len(md5_hash) != 32 or len([x for x in md5_hash if x in string.hexdigits]) != 32:
|
||||
raise Exception("Invalid replay hash")
|
||||
self.score_hash = md5_hash.encode()
|
||||
|
||||
|
||||
def set_mods(self, mods):
|
||||
if mods < 0 or mods > (1 << 30) - 1:
|
||||
raise Exception("Mods are out of range")
|
||||
self.mods = mods
|
||||
|
||||
|
||||
# --------------------------------------------------------------------------
|
||||
# Update variables ---------------------------------------------------------
|
||||
|
||||
def update_perfect(self):
|
||||
m = [self.score_100s, self.score_50s, self.score_katus, self.score_miss]
|
||||
self.perfect = sum(m) == 0
|
||||
|
||||
|
||||
def update_score_hash(self):
|
||||
self.set_score_hash(
|
||||
md5("%d%s%s%s%d%d" % (
|
||||
md5_str("%d%s%s%s%d%d" % (
|
||||
self.combo, "osu", self.player_name,
|
||||
self.beatmap_hash, self.score, self.get_rank()
|
||||
))
|
||||
)
|
||||
|
||||
|
||||
def update(self):
|
||||
self.update_perfect()
|
||||
self.update_score_hash()
|
||||
|
||||
|
||||
# --------------------------------------------------------------------------
|
||||
# Get / Helpers ------------------------------------------------------------
|
||||
|
||||
|
@ -131,7 +152,7 @@ class Replay:
|
|||
if r300 > .6:
|
||||
return osuRanks.C
|
||||
return osuRanks.D
|
||||
|
||||
|
||||
# --------------------------------------------------------------------------
|
||||
# Write replay data --------------------------------------------------------
|
||||
|
||||
|
@ -140,9 +161,9 @@ class Replay:
|
|||
for _frame in frame: # (frames)
|
||||
self.write(_frame)
|
||||
return
|
||||
|
||||
|
||||
self.replay_data += bytes(frame)
|
||||
|
||||
|
||||
# --------------------------------------------------------------------------
|
||||
# IO replay data -----------------------------------------------------------
|
||||
|
||||
|
|
Loading…
Reference in New Issue
Block a user