>>> import os
>>> current_env=os.environ.get("XDG_CURRENT_DESKTOP", '').split(":")
>>> print (current_env)
['Budgie', 'GNOME']
>>> if current_env in ('Unity', 'Budgie'):
...   print ("its in here")
... 
>>>