calculate 2 months from date in python code example

Example: python months between two dates

from datetime import datetime

def diff_month(d1, d2):
    return (d1.year - d2.year) * 12 + d1.month - d2.month