I am trying to match strings with an addition of 1 at the end of it and my code gives me this problem:

1. abcd01 should become abcde02 but my code gives me this output: abcde2
2. foobar00 should become foobar01 but my code gives me this output: foobar1
3. foobar001 should become foobar002
4. foobar1 should become foobar2 My code:

def increment_string(strng):regex = re.compile(r'[0-9]')match = regex.findall(strng)nums = ''.join(match[-3:])strng = strng.replace(nums,'')add = int(nums)+1print(strng+str(add))

Count the number of zeros until the first number and add them to the number at last. Here is the full code:

def increment_string(strng):regex = re.compile(r'[0-9]')match = regex.findall(strng)nums = ''.join(match[-3:])strng = strng.replace(nums, '')if len(nums) > 0:add = int(nums) + 1zeros = 0for digit in str(nums[:-1]):if digit == "0":zeros += 1else:breakzeros -= len(str(add)) - len(str(add-1))add = ''.join(["0" for x in range(zeros)]) + str(add)else:add = "1"print(strng + add)

Output:

abcd02

For foobar00:

increment_string('foobar00')

Output:

foobar01

For foobar002:

increment_string('foobar002')

Output:

foobar003

For abcd:

increment_string('abcd')

Output:

abcd1