FilesExpand file tree

 main

/

util.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

101 lines (85 loc) · 2.64 KB

 main

/

util.py

Copy path

Top

File metadata and controls

• Code
• Blame

101 lines (85 loc) · 2.64 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

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

96

97

98

99

100

101

ANSI_COLOR_RESET = "\x1b[0m"

ANSI_COLOR_YELLOW = "\x1b[33m"

ANSI_BOLD = "\x1b[1m"

ANSI_COLOR_GREEN = "\x1b[32m"

CONSOLE_COLOR_DEFAULT = ANSI_COLOR_RESET

CONSOLE_COLOR_PROMPT = ANSI_COLOR_YELLOW

CONSOLE_COLOR_USER_INPUT = ANSI_BOLD + ANSI_COLOR_GREEN

# Iterative search

# Actively searches and prevents a pattern from being returned

class IterSearch:

def __init__(self, pattern):

self.pattern = list(pattern)

self.buffer = []

def __call__(self, char):

self.buffer += [char]

if self.pattern[: len(self.buffer)] == self.buffer:

if len(self.buffer) >= len(self.pattern):

self.buffer.clear()

return []

_tmp = self.buffer[:]

self.buffer.clear()

return _tmp

class Circle:

def __init__(self, size, default=0):

self.list = [default] * size

self.maxsize = size

self.size = 0

self.offset = 0

def append(self, elem):

if self.size < self.maxsize:

self.list[self.size] = elem

self.size += 1

else:

self.list[self.offset] = elem

self.offset = (self.offset + 1) % self.maxsize

def __getitem__(self, val):

if isinstance(val, int):

if 0 > val or val >= self.size:

raise IndexError("Index out of range")

return (

self.list[val]

if self.size < self.maxsize

else self.list[(self.offset + val) % self.maxsize]

)

elif isinstance(val, slice):

start, stop, step = val.start, val.stop, val.step

if step is None:

step = 1

if start is None:

start = 0

if stop is None:

stop = self.size

if start < 0:

start = self.size + start

if stop < 0:

stop = self.size + stop

indices = range(start, stop, step)

return [

self.list[(self.offset + i) % self.maxsize]

for i in indices

if i < self.size

]

else:

raise TypeError("Invalid argument type")

if __name__ == "__main__":

c = Circle(5)

c.append(1)

print(c.list)

print(c[:])

assert c[0] == 1

assert c[:5] == [1]

for i in range(2, 5 + 1):

c.append(i)

print(c.list)

print(c[:])

assert c[0] == 1

assert c[:5] == [1, 2, 3, 4, 5]

for i in range(5 + 1, 9 + 1):

c.append(i)

print(c.list)

print(c[:])

assert c[0] == 5

assert c[:5] == [5, 6, 7, 8, 9]

# assert c[:-5] == [5,6,7,8,9]

assert c[:10] == [5, 6, 7, 8, 9]