WordPress :/

wordpress.py

import json

import pymysql

from django.conf import settings


class WordPress:
    db = None
    cursor = None

    def __init__(self):
        self.db = pymysql.connect(
            settings.MYSQL_HOST,
            settings.MYSQL_USER,
            settings.MYSQL_PASS,
            settings.MYSQL_DBNAME,
            cursorclass=pymysql.cursors.DictCursor)

        self.cursor = self.db.cursor()

    def get_user_id(self, email):
        sql = "SELECT `entry_id` FROM `JH_gf_entry_meta` WHERE `meta_value`=%s"
        self.cursor.execute(sql, email)
        row = self.cursor.fetchone()

        return row['entry_id']

    def get_user_info(self, user_id, trusted=False):
        sql = "SELECT `meta_key`, `meta_value` FROM `JH_gf_entry_meta` WHERE `entry_id`=%s"
        self.cursor.execute(sql, user_id)

        user_data = {
            'external_id': user_id
        }

        for meta_data in self.cursor.fetchall():
            if meta_data['meta_key'] == '3.3':
                user_data['first_name'] = meta_data['meta_value']

            if meta_data['meta_key'] == '3.6':
                user_data['last_name'] = meta_data['meta_value']

            if meta_data['meta_key'] == '8':
                user_data['email'] = meta_data['meta_value']

            if meta_data['meta_key'] == '31':
                user_data['mobile'] = meta_data['meta_value']

            if meta_data['meta_key'] == '5.1':
                user_data['address'] = meta_data['meta_value']

            if meta_data['meta_key'] == '5.3':
                user_data['city'] = meta_data['meta_value']

            if meta_data['meta_key'] == '5.5':
                user_data['postal'] = meta_data['meta_value']

            if meta_data['meta_key'] == '5.6':
                if meta_data['meta_value'] == 'Nederland':
                    user_data['country_code'] = 'NL'
                if meta_data['meta_value'] == 'België':
                    user_data['country_code'] = 'BE'
                if meta_data['meta_value'] == 'Duitsland':
                    user_data['country_code'] = 'DE'
                if meta_data['meta_value'] == 'Luxemburg':
                    user_data['country_code'] = 'LU'

            if trusted:
                if meta_data['meta_key'] == '36':
                    user_data['iban'] = meta_data['meta_value']

                if meta_data['meta_key'] == '37':
                    user_data['bsn'] = meta_data['meta_value']

            if meta_data['meta_key'] == '26':
                user_data['image'] = json.loads(meta_data['meta_value'])[0]

        return user_data