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
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
|
import datetime
import random
import string
class Generator:
UTC = datetime.timezone.utc
DATE_2018 = datetime.datetime(2018, 1, 1, tzinfo=UTC)
DATE_2023 = datetime.datetime(2023, 1, 1, tzinfo=UTC)
PERMISSIONS = ["READ", "WRITE", "READ_WRITE"]
NAMES = [
"alice",
"ben",
"carol",
"daniel",
"esther",
"franklin",
"genevieve",
"harold",
"ilana",
"jerome",
"katelyn",
"leonard",
"monique",
"nathan",
"ora",
"patrick",
"quinn",
"ronald",
"stephanie",
"thomas",
"uma",
"vince",
"wendy",
"xavier",
"yitzchak",
"zahra",
]
def __init__(self, capacity, seed=42):
self.capacity = capacity
self.random = random.Random(seed)
def randdt(self, min, max):
ts = self.random.randint(min.timestamp(), max.timestamp())
return datetime.datetime.fromtimestamp(ts).replace(tzinfo=self.UTC)
def randstr(self, min=None, max=None):
if max is not None:
min = self.random.randint(min, max)
return "".join(self.random.choices(string.ascii_letters, k=min))
def make(self, is_dir):
name = self.randstr(4, 30)
created_by = self.random.choice(self.NAMES)
created_at = self.randdt(self.DATE_2018, self.DATE_2023)
data = {
"type": "directory" if is_dir else "file",
"name": name,
"created_by": created_by,
"created_at": created_at.isoformat(),
}
if self.random.random() > 0.75:
updated_by = self.random.choice(self.NAMES)
updated_at = self.randdt(created_at, self.DATE_2023)
data.update(
updated_by=updated_by,
updated_at=updated_at.isoformat(),
)
if is_dir:
n = min(self.random.randint(0, 30), self.capacity)
self.capacity -= n
data["contents"] = [self.make_node() for _ in range(n)]
else:
data["nbytes"] = self.random.randint(0, 1000000)
data["permissions"] = self.random.choice(self.PERMISSIONS)
return data
def make_node(self):
return self.make(self.random.random() > 0.8)
def generate(self):
self.capacity -= 1
if self.capacity == 0:
out = self.make(False)
else:
out = self.make(True)
while self.capacity:
self.capacity -= 1
out["contents"].append(self.make_node())
return out
def make_filesystem_data(n):
return Generator(n).generate()
|
