flask-mongodb-fakultas-prodi-api/controller.py at main · satrio-pamungkas/flask-mongodb-fakultas-prodi-api · 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
from config import *
import pymongo
import json

client = pymongo.MongoClient(f"mongodb+srv://{DB_USERNAME}:{DB_PASSWORD}@cluster0.od9px.mongodb.net/{DB_NAME}?retryWrites=true&w=majority")
db = client[DB_NAME]
collection = db[DB_COLLECTION]

def get_all():
    base = {"universitas": "Universitas Pendidikan Indonesia"}
    cursor = collection.distinct('fakultas', {})

    list_fakultas = []
    for fakultas in cursor:
        data = {'namaFakultas': fakultas, 'listProdi': get_all_list(fakultas)}
        list_fakultas.append(data)

    base['listFakultas'] = list_fakultas
    return base


def get_all_list(nama_fakultas):
    cursor = collection.find({'fakultas': nama_fakultas}, {'kode':1, 'prodi':1, '_id':0})

    if cursor.count() == 0:
        return None

    list_prodi = []
    for prodi in cursor:
        data = {'kodeProdi': prodi['kode'], 'namaProdi': prodi['prodi']}
        list_prodi.append(data)

    return list_prodi


def get_list_fakultas():
    cursor = collection.distinct('fakultas', {})

    if cursor is None:
        return None

    list_fakultas = []
    for fakultas in cursor:
        data = {'namaFakultas': fakultas}
        list_fakultas.append(data)

    return list_fakultas


def get_list_prodi_fakultas(nama_fakultas):
    base = {"namaFakultas": nama_fakultas}
    cursor = collection.find({'fakultas': nama_fakultas}, {'kode':1, 'prodi':1, '_id':0})

    if cursor.count() == 0:
        return None

    list_prodi = []
    for prodi in cursor:
        data = {'kodeProdi': prodi['kode'], 'namaProdi': prodi['prodi']}
        list_prodi.append(data)

    base['listProdi'] = list_prodi
    return base


def get_list_prodi():
    cursor = collection.find({}, {'kode':1, 'prodi':1, '_id':0})

    if cursor is None:
        return None

    list_prodi = list(cursor)
    return list_prodi


def get_prodi(kode_prodi):
    cursor = collection.find_one({'kode': kode_prodi}, {'_id':0})

    if cursor is None:
        return None

    return cursor