user_dep.py
8.52 KB
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
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
import colander
import os
import logging
from deform import widget
from pyramid.view import view_config
from . import BaseView, button_import
from ...models import User, DepartemenUser, Departemen, DBSession
from opensipkd.tools.report import open_rml_row, csv_response, open_rml_pdf, pdf_response, file_response
from .departemen import get_departemen_list
from .user import user_list
from pyramid.i18n import TranslationStringFactory
from pyramid.path import AssetResolver
from . import BaseView, button_import, get_params
_ = TranslationStringFactory('myapp')
log = logging.getLogger(__name__)
@colander.deferred
def departemen_widget(node, kw):
values = kw.get('departemen_list', [])
return widget.CheckboxChoiceWidget(values=values, multiple=True, placeholder='Pilih Departemen')
class ListSchema(colander.Schema):
id = colander.SchemaNode(
colander.Integer(),
title="Action"
)
user_name = colander.SchemaNode(
colander.String(),
field=User.user_name,
title="User"
)
departemen_name = colander.SchemaNode(
colander.String(),
field=Departemen.nama,
title="Departemen"
)
@colander.deferred
def departemen_checkbox_widget(node, kw):
values = kw.get('departemen_list', [])
return widget.CheckboxChoiceWidget(values=values, multiple=True)
def validate_departemen_required(node, value):
"""Validator untuk memastikan setidaknya satu departemen dipilih"""
if not value or len(value) == 0:
raise colander.Invalid(node, "Departemen wajib diisi. Pilih setidaknya satu departemen.")
class AddSchema(colander.Schema):
user_id = colander.SchemaNode(
colander.Integer(),
widget=widget.SelectWidget(values=User.get_list()),
oid="user_id",
title="User",
)
departemen_id = colander.SchemaNode(
colander.Set(),
widget=departemen_checkbox_widget,
oid="departemen_id",
title="Departemen",
validator=validate_departemen_required # Tambahkan validator wajib
)
class EditSchema(AddSchema):
id = colander.SchemaNode(
colander.String(),
missing=colander.drop,
widget=widget.HiddenWidget()
)
# Override departemen_id agar tidak wajib saat edit
departemen_id = colander.SchemaNode(
colander.Set(),
widget=departemen_checkbox_widget,
oid="departemen_id",
title="Departemen",
validator=None # Tidak wajib saat edit
)
class Views(BaseView):
def __init__(self, request):
super().__init__(request)
self.request = request
self.list_schema = ListSchema
self.add_schema = AddSchema
self.edit_schema = EditSchema
self.list_route = 'user-departemen'
self.table = DepartemenUser
self.list_buttons = self.list_buttons + self.list_report + (button_import,)
path = os.path.dirname(__file__)
path = os.path.dirname(path)
self.report_file = os.path.join(path, 'reports', 'userdep1.jrxml')
def list_join(self, query, **kwargs):
return query.outerjoin(Departemen, Departemen.id == self.table.departemen_id) \
.outerjoin(User, User.id == self.table.user_id)
@view_config(route_name='user-departemen', renderer='templates/table.pt',
permission='user-view')
def view_list(self, **kwargs):
return super().view_list(**kwargs)
def get_module_path(self, module_name):
a = AssetResolver(module_name)
resolver = a.resolve('')
return resolver.abspath()
def pdf_response(self, **kwargs):
# print("PDF RESPONSE======================================")
from opensipkd.base.tools.report import jasper_export
logo = self.get_module_path('opensipkd.base')
logo = os.path.join(logo, 'static', 'img', 'logo.png')
parameters = {
"judul": get_params('company', "openSIPKD"),
"logo": logo
}
print(parameters)
filename = jasper_export(self.report_file, parameters=parameters)
return file_response(self.req, filename=filename[0])
@view_config(route_name='user-departemen-act', renderer='json',
permission='user-view')
def view_act(self):
return super().view_act()
def get_bindings(self, row=None):
bindings = {
"departemen_list": get_departemen_list(),
"user_list": user_list()
}
if row and hasattr(row, 'user_id'):
existing_depts = self.get_existing_departemen(row.user_id)
bindings["default_values"] = {
"user_id": row.user_id,
"departemen_id": list(existing_depts)
}
return bindings
@view_config(route_name='user-departemen-add', renderer='templates/form.pt',
permission='user-edit')
def view_add(self):
return super().view_add()
@view_config(route_name='user-departemen-view', renderer='templates/form.pt',
permission='user-view')
def view_view(self):
return super().view_view()
@view_config(route_name='user-departemen-delete', renderer='templates/form.pt',
permission='user-edit')
def view_delete(self):
return super().view_delete()
@view_config(route_name='user-departemen-edit', renderer='templates/form.pt',
permission='user-edit')
def view_edit(self):
return super().view_edit()
@view_config(route_name='user-departemen-upload',
renderer='templates/form.pt', permission='user-upload')
def view_upload(self):
self.upload_keys = ["user_id"]
return super(Views, self).view_upload(exts=(".csv", ".tsv"))
def save_request(self, values, row=None):
user_id = values.get("user_id")
if not user_id:
return None
if "departemen_id" in values:
selected_departemen = set(
str(departemen_id) for departemen_id in values["departemen_id"]
if departemen_id is not None and str(departemen_id) != "None"
)
# Batasi ke maksimal dua pilihan
selected_list = sorted(selected_departemen)[:2]
dept1 = int(selected_list[0]) if selected_list else None
dept2 = int(selected_list[1]) if len(selected_list) > 1 else None
# Cari entri yang sudah ada untuk user_id
existing_row = DBSession.query(DepartemenUser).filter_by(user_id=user_id).first()
if row and row.user_id != user_id:
# Jika user_id berubah, hapus entri lama dan buat baru
DBSession.query(DepartemenUser).filter_by(user_id=row.user_id).delete()
if selected_list:
new_row = DepartemenUser(
user_id=user_id,
departemen_id=dept1,
sub_departemen=dept2
)
DBSession.add(new_row)
row = new_row
else:
row = None
else:
# Jika user_id sama atau ini adalah tambah baru
if existing_row:
# Update entri yang sudah ada jika user_id sama
if selected_list:
existing_row.departemen_id = dept1
existing_row.sub_departemen = dept2
# Jika selected_list kosong, biarkan data lama tetap ada
row = existing_row
else:
# Buat entri baru jika belum ada
if selected_list: # Ini akan selalu ada saat tambah karena validator
new_row = DepartemenUser(
user_id=user_id,
departemen_id=dept1,
sub_departemen=dept2
)
DBSession.add(new_row)
row = new_row
else:
row = None
DBSession.flush()
return row
def get_existing_departemen(self, user_id):
row = DBSession.query(DepartemenUser).filter_by(user_id=user_id).first()
result = set()
if row:
if row.departemen_id:
result.add(str(row.departemen_id))
if row.sub_departemen:
result.add(str(row.sub_departemen))
return result
def get_values(self, row, istime=False):
d = super(Views, self).get_values(row, istime)
d["departemen_id"] = self.get_existing_departemen(row.user_id)
return d