"""User model """ from bulkperformancerights.connectors import mysql def user_information(user_id, user_type='vendor'): """User information Returns a user's logo path and client manager email Args: (int): user_id (str): user_type Return: (list): track ids owned by the specified user """ session = mysql.art_relations_session() vendor_image_sql = user_information_sql(user_id, user_type) user_info = session.execute(vendor_image_sql).fetchone() session.close() return user_info def user_information_sql(user_id, user_type='vendor'): """SQL to get information for a user Args: (int): user_id (str): user_type Return: (str): SQL for user's information """ sql = ''' SELECT ia.path, ia.filename, oau.email as vendor_contact_email FROM vendor v LEFT JOIN subaccount s ON s.vendor_id = v.vendor_id INNER JOIN orchadmin_users oau ON v.assigned_to = oau.id LEFT JOIN vendor_logo vl on vl.vendor_id = v.vendor_id LEFT JOIN image_assets ia ON ia.id = vl.image_asset_id WHERE {ownership_clause} ORDER BY date_added DESC LIMIT 1;''' if user_type == 'subaccount': ownership_clause = 's.subaccount_id = {}'.format( user_id) else: ownership_clause = 'v.vendor_id = {}'.format( user_id) return sql.format(ownership_clause=ownership_clause)