from io.hevo.api import Event
import json
import hashlib
from phoneiso3166.country import phonecountry, InvalidPhone
# prepare email string and calculate sha256
def emailhash(mes):
   if mes:
       try:
           prep = mes.strip().lower()
           return hashlib.sha256(prep).hexdigest()
       except:
           return None
# extract email domain from address   
def emaildomain(mes):
   if mes:
       try:
           prep = mes.split('@')[1]
           return prep
       except:
           return None
# extract country code from phone number
def phonecountrycode(phonenumber):
   try:
       return phonecountry(phonenumber)
   except:
       return None
# main transformation handler
def transform(event):
   # Get event name & its properties
   eventName = event.getEventName()
   properties = event.getProperties()
   if eventName == 'users':
       # calculate hash values
       properties['emailsha256'] = emailhash(properties.get('email'))
       properties['emaildomain'] = emaildomain(properties.get('email'))
       properties['phonecountry'] = phonecountrycode(properties.get('phone'))
       # drop pii for all events
       pii = ['email', 'firstname', 'lastname', 'phone', 'pendingphone', 'hashedpassword', 'photo_url', 'keywords']
       [properties.pop(p, None) for p in properties.keys() if p in pii]
   return event