Art Cinema 프로젝트 개발일지4

고객DB 예시 만들기

앞서 메인 알고리즘을 구현했고,
그다음 해야할 기능은 "고객이 선호하는 메인장르와 서브장르에 맞는 단편영화들을 추천"하는 기능이다.

이 기능을 구현하기 앞서, 메인 알고리즘에 따른 가상의(?) 고객 DB를 구현하는 코드를 짰다.

앞에 개발일지3번에서 했던 메인 코드를 함수로 묶어주었다.

def movie_select():

그리고 가상의 20명의 고객 의 DB를 만들기 위해서,
for문이 20번 돌아가게 만들어줬다.

for문안에는 먼저 함수 실행으로 메인알고리즘이 실행되고, 그때마다 데이터들이 저장되게 만들었다.

for k in range(20):
    #고객의 선호조사 함수
    movie_select()
    #임의의 고객정보 저장
    customer_name = "스파르타" + str(k)
    customer_Id = "sparta" + str(k)
    customer_genre1 = result_1
    customer_genre2 = result_2

    #고객데이터
    customer_info = {
                     'name': customer_name,
                     'Id': customer_Id,
                     'main_genre': customer_genre1,
                     'second_genre': customer_genre2,
    }
    #Customer 이라는 db에 고객정보들 저장
    db.Customer.insert_one(customer_info)

변수 name 과 Id는 임의로 설정해 주었고,
변수 customer_genre1 과 customer_genre2에
함수 main_select()에서 임의로 설정된 메인 장르와 서브장르가 들어가게 된다.

그리고 이를 Customer 이라는 db에 저장해주고,
20번 돌아가기 때문에,
sparta0번 부터 sparta19번 까지의 고객의 db가 저장된다.