正则表达式，匹配A与B之间的部分，如果没有B，取A之后的部分

Question

先说效果：
AVVBCCC 匹配出 VV
ACCCCCD 匹配出CCCCCD

单独实现我都会
第一种

# -*- coding: utf-8 -*-import re#str = 'AVVBCCC'forword = re.search(r'A(.*)B', str).group(1)
print(forword)

第二种

# -*- coding: utf-8 -*-import re#str = 'ACCCCCD'forword = re.search(r'A(.*)', str).group(1)
print(forword)

请问怎么合起来写？

Answer 1

先判断下字符串中有没有B，有就第一种没有就第二种