/
object_manager.py
52 lines (43 loc) · 1.42 KB
/
object_manager.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
import pickle
from data_manager import data_manager
import random
import os
class object_manager():
data_file_name = 'just_some_data'
def __init__(self, data_file):
self.data_file_name=data_file
def object_create(self):
dm = data_manager(self.data_file_name)
values = dm.unpickle_data()
values["counter"] = values["counter"] + 1
values["obj"].append(int(values["counter"]))
dm.pickle_data(values)
return True
def object_reset(self):
dm = data_manager(self.data_file_name)
objs = []
dm.pickle_data({"obj": objs, "counter": len(objs)})
return True
def object_get(self):
dm = data_manager(self.data_file_name)
values = dm.unpickle_data()
obj = None
if len(values["obj"]) > 0:
obj = random.choice(values["obj"])
values["obj"].remove(obj)
dm.pickle_data(values)
return obj
def object_list(self):
dm = data_manager(self.data_file_name)
values = dm.unpickle_data()
return values["obj"]
def object_free(self,num):
dm = data_manager(self.data_file_name)
values = dm.unpickle_data()
values["obj"].append(int(num))
values["obj"].sort()
dm.pickle_data(values)
return True
def delete_data_file(self):
if os.path.exists(self.data_file_name):
os.remove(self.data_file_name)