AI-GP-V1A-Projectgroep_6/popular_return_recoms.py at main · beaupd/AI-GP-V1A-Projectgroep_6 · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
from rdbconnection2 import conrdb
from collections import Counter

rdbcon, rdbcur = conrdb()

def return_selection(sql_query, query_var):
    """
        Voert een query uit, met de meegegeven variabelen.
        (Gaat uit van een 'SELECT' query waarbij data terug moet worden gehaald.)
        Geeft de geselecteerde waardes terug.

        :param sql_query: str: query die uitgevoerd moet worden
        :param query_var: tuple: variabelen die nodig zijn om de query uit te voeren
        :returns: opgevraagde element
    """
    rdbcon, rdbcur = conrdb()
    rdbcur.execute(sql_query, query_var)
    returnvalue = []
    try:
        returnvalue = rdbcur.fetchone()[0]
        rdbcur.close()
        rdbcon.close()
    except TypeError:
        pass
    return returnvalue

def highest_frequency(array):
    """
        Zoekt het element met de hoogste frequentie in een lijst.

        :param array: list: lijst met elementen
        :returns: element met hoogste freq
    """
    freq = Counter(array)
    return freq.most_common(1)[0][0]

def simple_algorithm(category_page):
    """
        Deze functie geeft recommendations
        uit de meest_gekocht tabel filter op basis van de categoriepagina.

        :param category_page: str : categorie
        :returns: list: lijst met recommended productid's
    """
    prodids = []
    category_bestaat_query = """select exists(select * from meest_gekocht where category=%s)::int;"""
    if return_selection(category_bestaat_query, (category_page,)):
        prodids_query = """select array[product_id1, product_id2, product_id3, product_id4] from meest_gekocht where category=%s;"""
        prodids = return_selection(prodids_query, (category_page,))
        return prodids

def gender_category_based_algorithm(klik_events, category_page):
    """
        Deze functie geeft recommendations
        uit de gender_category tabel filter op basis van het meestvoorkomende
        gender en de huidige categoriepagina.

        :param klik_events: list: lijst met productid's
        :param category_page: str : categorie
        :returns: list: lijst met recommended product_id's
    """
    prodids = []
    user_all_genders_query = """select array(select lower(gender) from product where product_id=ANY(%s::varchar[]));"""
    user_all_genders = return_selection(user_all_genders_query, (klik_events,))
    user_gender = highest_frequency(user_all_genders)
    if user_gender is None:
        _combi_bestaat_query = """select exists(select * from gender_category where gender is Null and category=%s)::int"""
        if return_selection(_combi_bestaat_query, (category_page,)):
            prodids_query = """select array[product_id1, product_id2, product_id3, product_id4] from gender_category where gender is Null and category=%s"""
            prodids = return_selection(prodids_query, (category_page,))
    else:
        _combi_bestaat_query = """select exists(select * from gender_category where gender=%s and category=%s)::int"""
        if return_selection(_combi_bestaat_query, (user_gender, category_page,)):
            prodids_query = """select array[product_id1, product_id2, product_id3, product_id4] from gender_category where gender=%s and category=%s"""
            prodids = return_selection(prodids_query, (user_gender, category_page,))
    return prodids

def return_recommended_products(profileid, category_page, klik_events):
    """
        Deze functie geeft een lijst met productid's terug die
        passen bij het gedrag van de user op de website.

        :param profileid: str : profiel van user
        :param category_page: str: categorie
        :param klikevents: list: lijst met productid's
        :returns: list : lijst met productid's
    """
    prodids = []
    _user_is_bekend_query = """select exists (select * from profile where profile_id=%s)::int"""
    _user_heeft_orders_query = """select exists(select * from orders
                                    natural join(select * from sessions
                                    natural join(select browser_id from buids
                                    where profile_id=%s) as _id) as __id)::int;"""
    if return_selection(_user_is_bekend_query, (profileid,)) and return_selection(_user_heeft_orders_query, (profileid,)):
        user_segment_query = """select lower(segment) from profile where profile_id=%s"""
        user_segment = return_selection(user_segment_query, (profileid,))
        user_all_genders_query = """select array(select lower(gender) from product
                                    natural join(select product_id from orders
                                    natural join(select session_id from sessions
                                    natural join(select browser_id from buids
                                    where profile_id=%s) as _id) as __id) as ___id)"""
        user_all_genders = return_selection(user_all_genders_query, (profileid,))
        user_gender = highest_frequency(user_all_genders)
        if user_gender is None:
            _combi_bestaat_query = """select exists(select * from popular where lower(segment)=%s and gender is Null and lower(category)=%s)::int;"""
            if return_selection(_combi_bestaat_query, (user_segment, category_page,)):
                prodids_query = """select array[product_id1, product_id2, product_id3, product_id4] from popular
                                    where lower(segment)=%s and gender is Null and lower(category)=%s;"""
                prodids = return_selection(prodids_query, (user_segment, category_page,))
        else:
            _combi_bestaat_query = """select exists(select * from popular where lower(segment)=%s and lower(gender)=%s and lower(category)=%s)::int;"""
            if return_selection(_combi_bestaat_query, (user_segment, user_gender, category_page,)):
                prodids_query = """select array[product_id1, product_id2, product_id3, product_id4] from popular
                                    where lower(segment)=%s and lower(gender)=%s and lower(category)=%s;"""
                prodids = return_selection(prodids_query, (user_segment, user_gender, category_page,))
    else:
        if klik_events:
            prodids = gender_category_based_algorithm(klik_events, category_page)
        else:
            prodids = simple_algorithm(category_page)
    return prodids