1
#!/usr/bin/python
2
import re
3
import sys
4

5
#v0.9.6
6
fileName="listener_test.yml"
7
dir="./unregister"
8
variable="bar"
9
block_start='CodeUri: ' + dir
10
block_end='AWS::Serverless::Function'
11
rtext = '      AutoPublishCodeSha256: ' + variable + '\n'
12

13
with open("listener_test.yml") as ofile:
14
      lines=ofile.readlines()
15
      i = len(lines) - 1
16
      AWSFound = False
17
      CodeUriFound = False
18
      AutoFound = False
19
      unum = 0
20
      while i >= 0 and not AWSFound:
21
           if block_start in lines[i]:
22
             CodeUriFound = True
23
             unum = i
24
           if "AutoPublishCodeSha256:" in lines[i]:
25
             AutoFound = True
26
             unum = i
27
           if block_end in lines[i] and CodeUriFound:
28
             AWSFound = True
29

30
           i -= 1
31

32
if AutoFound:
33
  lines[unum] = rtext
34
else:
35
  lines.insert(unum - 1, rtext)
36

37
with open('listener_test_new.yml', 'w') as file:
38
  lines = "".join(lines)
39
  file.write(lines)