Fakultas Ilmu Komputer UI

accounts.py 4.49 KB
Newer Older
1
2
3
4
5
import os
import uuid

from django.contrib.auth.models import User
from django.core.validators import MinValueValidator, MaxValueValidator
6
from django.db import models
7

8
9
from core.lib.validators import validate_document_file_extension, validate_image_file_extension

10
11

def get_student_resume_file_path(instance, filename):
12
13
14
15
16
    extension = filename.split('.')[-1].lower()
    filename = "%s.%s" % (uuid.uuid4(), extension)
    return os.path.join("student-resume/", filename)


17
18
19
20
21
22
def get_student_photo_file_path(instance, filename):
    extension = filename.split('.')[-1].lower()
    filename = "%s.%s" % (uuid.uuid4(), extension)
    return os.path.join("student-photo/", filename)


23
24
25
26
27
28
def get_company_logo_file_path(instance, filename):
    extension = filename.split('.')[-1].lower()
    filename = "%s.%s" % (uuid.uuid4(), extension)
    return os.path.join("company-logo/", filename)


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
def get_display_name(user, full_name=False):
    """
    Return a display name that always works like "Benoit J."
    """
    if user.first_name and user.last_name:
        if full_name:
            last = user.last_name.title()
        else:
            last = user.last_name[:1].upper() + "."
        name = user.first_name.title() + " " + last
    elif user.first_name:
        name = user.first_name.title()
    elif user.last_name:
        name = user.last_name.title()
    else:
        name = user.username

    return name.strip()


class Student(models.Model):
    """
    A user subclass
    """
    created = models.DateTimeField(auto_now_add=True)
    updated = models.DateTimeField(auto_now=True)
    user = models.OneToOneField(User)
    npm = models.IntegerField(validators=[MinValueValidator(100000000), MaxValueValidator(9999999999)], unique=True)
57
    resume = models.FileField(upload_to=get_student_resume_file_path, null=True, blank=True, validators=[validate_document_file_extension])
58
    phone_number = models.CharField(max_length=100, blank=True, db_index=True, null=True)
59
    bookmarked_vacancies = models.ManyToManyField('core.Vacancy', related_name="bookmarked_vacancies", blank=True)
60
61
    applied_vacancies = models.ManyToManyField('core.Vacancy', related_name="applied_vacancies",
                                               blank=True, through='core.Application')
62
63
64
65
    birth_place = models.CharField(max_length=30, blank=True, null=True)
    birth_date = models.DateField(blank=True, null=True)
    major = models.CharField(max_length=30, blank=True, null=True)
    batch = models.CharField(max_length=4, blank=True, null=True)
66
    show_transcript = models.BooleanField(default=False)
67
    photo = models.FileField(upload_to=get_student_photo_file_path, null=True, blank=True, validators=[validate_image_file_extension])
68
    linkedin_url = models.URLField(max_length=200,blank=True, null=True)
PRIAMBUDI LINTANG BAGASKARA's avatar
PRIAMBUDI LINTANG BAGASKARA committed
69
    region = models.CharField(max_length=30, blank=True, null=True)
70

Zamil Majdy's avatar
Zamil Majdy committed
71
72
73
74
    @property
    def name(self):
        return get_display_name(self.user)

75
76
77
78
    @property
    def full_name(self):
        return get_display_name(self.user, full_name=True)

79
80
81
    def __unicode__(self):
        return u"Student {}".format(get_display_name(self.user))

82
83
84
    class Meta:
        ordering = ['-updated']

85
86
87
88
89

class Company(models.Model):
    """
    A user subclass
    """
90
91
92
93
    NEW = 0
    UNVERIFIED = 1
    VERIFIED = 2

94
95
96
97
    created = models.DateTimeField(auto_now_add=True)
    updated = models.DateTimeField(auto_now=True)
    user = models.OneToOneField(User)
    description = models.TextField()
98
    status = models.IntegerField(default=NEW)
99
    logo = models.FileField(upload_to=get_company_logo_file_path, null=True, blank=True,  validators=[validate_image_file_extension])
100
    address = models.CharField(max_length=1000, blank=True, null=True)
101
    category = models.CharField(max_length=140, default="Belum ada kategori perusahaan")
102

Zamil Majdy's avatar
Zamil Majdy committed
103
104
105
106
    @property
    def name(self):
        return get_display_name(self.user)

107
108
109
    def __unicode__(self):
        return u"Company {}".format(get_display_name(self.user))

110
111
112
    class Meta:
        ordering = ['-updated']

113
114
115
116
117
118
119
120
121
122

class Supervisor(models.Model):
    """
    A user subclass
    """
    created = models.DateTimeField(auto_now_add=True)
    updated = models.DateTimeField(auto_now=True)
    user = models.OneToOneField(User)
    nip = models.IntegerField(validators=[MinValueValidator(100000000), MaxValueValidator(9999999999)], unique=True)

Zamil Majdy's avatar
Zamil Majdy committed
123
124
125
126
    @property
    def name(self):
        return get_display_name(self.user)

127
128
    def __unicode__(self):
        return u"Supervisor {}".format(get_display_name(self.user))
129
130
131

    class Meta:
        ordering = ['-updated']