PCPGR/areas.py at main · TuloShark/PCPGR · 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
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141

class areas:
    """
    This class represents the areas of the company.
    """

    areaDesarrollo = []
    areaContabilidad = []
    areaDisenno = []
    areaMarketing = []
    maxDesarrolladores = 0
    maxDiseno = 0
    maxMarketing = 0
    maxContabilidad = 0

    def __init__(self, maxAreaDis, maxAreaDevs, maxAreaMarke, maxAreaContabilidad):
        """
        Initializes the areas of the company.

        Parameters:
        maxAreaDis (int): Maximum number of developers in the design area.
        maxAreaDevs (int): Maximum number of developers in the development area.
        maxAreaMarke (int): Maximum number of developers in the marketing area.
        maxAreaContabilidad (int): Maximum number of developers in the accounting area.
        """
        self.maxDesarrolladores = maxAreaDevs
        self.maxDiseno = maxAreaDis
        self.maxMarketing = maxAreaMarke
        self.maxContabilidad = maxAreaContabilidad

    def obtenerDatosArea(self, area):
        """
        Returns the users in the specified area.

        Parameters:
        area (int): Area to retrieve the users from.

        Returns:
        list: List of users in the specified area.
        """
        if(area == 1):
            return self.areaDesarrollo
        elif(area == 2):
            return self.areaDisenno
        elif (area == 3):
            return self.areaMarketing
        elif (area == 4):
            return self.areaContabilidad
    def registrarUsuario(self, usuario, area):
        """
        Registers a user in the specified area.

        Parameters:
        usuario (object): User to be registered.
        area (int): Area to register the user in.

        Returns:
        int: 1 if the user was successfully registered, -1 if the area is full.
        """
        if (area== 1):
            if(self.maxDesarrolladores == 0):
                return -1
            self.areaDesarrollo.append(usuario)
            self.maxDesarrolladores-= 1
            return 1
        elif(area == 2):
            if(self.maxDiseno == 0):
                return -1
            self.areaDisenno.append(usuario)
            self.maxDiseno-= 1
            return 1
        elif (area == 3):
            if(self.maxMarketing == 0):
                return -1
            self.areaMarketing.append(usuario)
            self.maxMarketing-= 1
            return 1
        elif (area == 4):
            if(self.maxContabilidad == 0):
                return -1
            self.areaContabilidad.append(usuario)
            self.maxContabilidad-= 1
            return 1
    def consultarDisponibles(self,area):
        """
        Returns the number of available developers in the specified area.

        Parameters:
        area (int): Area to retrieve the available developers from.

        Returns:
        int: Number of available developers in the specified area.
        """
        if(area == 1):
            return self.maxDesarrolladores
        elif(area == 2):
            return self.maxDiseno
        elif (area == 3):
            return self.maxMarketing
        elif (area == 4):
            return self.maxContabilidad
    def eliminarUsuario(self, idUsuario, area):
        """
        Removes a user from the specified area.

        Parameters:
        idUsuario (int): ID of the user to be removed.
        area (int): Area to remove the user from.

        Returns:
        int: 1 if the user was successfully removed, -1 if the user was not found.
        """
        idUsuario = int(idUsuario)
        if area == 1:
            for usuario in self.areaDesarrollo:
                if int(usuario.getId()) == idUsuario:
                    self.areaDesarrollo.remove(usuario)
                    self.maxDesarrolladores+= 1
                    return 1
            return -1
        elif area == 2:
            for usuario in self.areaDisenno:
                if int(usuario.getId()) == idUsuario:
                    self.areaDisenno.remove(usuario)
                    self.maxDiseno+= 1
                    return 1
            return -1
        elif area == 3:
            for usuario in self.areaMarketing:
                if int(usuario.getId()) == idUsuario:
                    self.areaMarketing.remove(usuario)
                    self.maxMarketing+= 1
                    return 1
            return -1
        elif area == 4:
            for usuario in self.areaContabilidad:
                if int(usuario.getId()) == idUsuario:
                    self.areaContabilidad.remove(usuario)
                    self.maxContabilidad += 1
                    return 1
            return -1