python code for getting screen number

import os

def get\_screen\_info():

    screen\_id = os.environ.get("STY", None)
    if screen\_id:

        screen\_num = os.environ.get("WINDOW", None)

        return screen\_id, screen\_num

    return None, None