import imaplib
import email
import re


def get_code():
mail = imaplib.IMAP4_SSL('imap.gmail.com')
mail.login('manuel.berger00@gmail.com', '')
mail.list()
# Out: list of "folders" aka labels in gmail.
mail.select("Energy") # connect to inbox.

result, data = mail.search(None, "ALL")

ids = data[0] # data is a list.
id_list = ids.split() # ids is a space separated string
latest_email_id = id_list[-1] # get the latest

result, data = mail.fetch(latest_email_id, "(RFC822)") # fetch the email body (RFC822) for the given ID

raw_email = data[0][1] # here's the body, which is raw text of the whole email
# including headers and alternate payloads
#print result
#print raw_email
email_message = email.message_from_string(raw_email)

#print email_message['To']
#print email_message['From']
#print email_message['Subject']

#print the body

if email_message.is_multipart():
   for part in email_message.get_payload():
       print part.get_payload(decode=True).decode('utf-8')
else:
   print email_message.get_payload(decode=True).decode('utf-8')
   string = email_message.get_payload(decode=True).decode('utf-8')
   print re.sub('[^0-9]', '', string)[0:4]
   return re.sub('[^0-9]', '', string)[0:4]