diff --git a/yuthon_hr_attendance/controllers/applet_attendance.py b/yuthon_hr_attendance/controllers/applet_attendance.py index b145c0f8a..c4b8c7adc 100644 --- a/yuthon_hr_attendance/controllers/applet_attendance.py +++ b/yuthon_hr_attendance/controllers/applet_attendance.py @@ -849,12 +849,25 @@ class AppletAttendance(Controller): 'code': 400, 'message': '参数不完整' }) - request.env['wifi.message'].sudo().create({ - 'mac': mac, - 'name': name, - 'bssid': bssid, - 'company_id': company_id - }) + + wifi_message = request.env['wifi.message'].sudo().search([ + ('mac', '=', mac), + ('bssid', '=', bssid), + ('company_id', '=', company_id) + ], limit=1) + + if wifi_message: + wifi_message.sudo().write({ + 'name': name, + }) + else: + request.env['wifi.message'].sudo().create({ + 'mac': mac, + 'name': name, + 'bssid': bssid, + 'company_id': company_id + }) + return json.dumps({ 'code': 200, 'message': '成功' @@ -885,14 +898,27 @@ class AppletAttendance(Controller): 'code': 400, 'message': '参数不完整' }) - request.env['applet.address'].sudo().create({ - 'address_name': address_name, - 'address_name2': address_name2, - 'longitude': longitude, - 'latitude': latitude, - 'card_rice': card_rice, - 'company_id': company_id - }) + applet_address = request.env['applet.address'].sudo().search([ + ('address_name', '=', address_name), + ('longitude', '=', longitude), + ('latitude', '=', latitude), + ('company_id', '=', company_id) + ], limit=1) + + if applet_address: + applet_address.sudo().write({ + 'address_name2': address_name2, + 'card_rice': card_rice, + }) + else: + request.env['applet.address'].sudo().create({ + 'address_name': address_name, + 'address_name2': address_name2, + 'longitude': longitude, + 'latitude': latitude, + 'card_rice': card_rice, + 'company_id': company_id + }) return json.dumps({ 'code': 200, 'message': '成功'