Skip to content

Commit

Permalink
Apply ruff fix and format
Browse files Browse the repository at this point in the history
  • Loading branch information
153957 committed Oct 7, 2023
1 parent 99f1e53 commit 7835393
Show file tree
Hide file tree
Showing 42 changed files with 393 additions and 353 deletions.
4 changes: 2 additions & 2 deletions manage.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,8 +2,8 @@
import os
import sys

if __name__ == "__main__":
os.environ.setdefault("DJANGO_SETTINGS_MODULE", "publicdb.settings")
if __name__ == '__main__':
os.environ.setdefault('DJANGO_SETTINGS_MODULE', 'publicdb.settings')

from django.core.management import execute_from_command_line

Expand Down
22 changes: 11 additions & 11 deletions publicdb/analysissessions/models.py
Original file line number Diff line number Diff line change
Expand Up @@ -69,7 +69,7 @@ class AnalyzedCoincidence(models.Model):
error_estimate = models.FloatField(null=True, blank=True)

def __str__(self):
return f"{self.coincidence} - {self.student}"
return f'{self.coincidence} - {self.student}'

class Meta:
verbose_name = 'Analyzed coincidence'
Expand Down Expand Up @@ -100,7 +100,7 @@ class Meta:

@property
def name(self):
return f"{self.first_name} {self.sur_name}"
return f'{self.first_name} {self.sur_name}'

def create_session(self):
self.session_pending = False
Expand Down Expand Up @@ -157,7 +157,7 @@ def find_coincidence(self, date, session):
all_coincidences = cq.any(stations)
coincidences = cq.events_from_stations(all_coincidences, stations, n=3)
for coincidence in coincidences:
# Todo: Double check for multiple events from same station,
# TODO: Double check for multiple events from same station,
self.save_coincidence(coincidence, session)
number_of_coincidences += 1

Expand Down Expand Up @@ -212,14 +212,14 @@ def generate_url(self):
def sendmail_request(self):
subject = 'HiSPARC analysis session request'
message = textwrap.dedent(
f'''\
f"""\
Hello {self.name},
Please click on this link to confirm your request for an analysis session with jSparc:
https://data.hisparc.nl/analysis-session/request/{self.url}/
Greetings,
The HiSPARC Team'''
The HiSPARC Team""",
)
sender = 'Beheer HiSPARC <[email protected]>'
send_mail(subject, message, sender, [self.email], fail_silently=False)
Expand All @@ -229,7 +229,7 @@ def sendmail_request(self):
def sendmail_created(self):
subject = 'HiSPARC analysis session created'
message = textwrap.dedent(
f'''\
f"""\
Hello {self.name},
Your analysis session for jSparc has been created.
Expand All @@ -244,15 +244,15 @@ def sendmail_created(self):
https://data.hisparc.nl/analysis-session/{slugify(self.sid)}/data/
Greetings,
The HiSPARC Team'''
The HiSPARC Team""",
)
sender = 'Beheer HiSPARC <[email protected]>'
send_mail(subject, message, sender, [self.email], fail_silently=False)

def sendmail_created_less(self):
subject = 'HiSPARC analysis session created with less events'
message = textwrap.dedent(
f'''\
f"""\
Hello {self.name},
Your analysis session for jSparc has been created.
Expand All @@ -269,23 +269,23 @@ def sendmail_created_less(self):
https://data.hisparc.nl/analysis-session/{slugify(self.sid)}/data/
Greetings,
The HiSPARC Team'''
The HiSPARC Team""",
)
sender = 'Beheer HiSPARC <[email protected]>'
send_mail(subject, message, sender, [self.email], fail_silently=False)

def sendmail_zero(self):
subject = 'HiSPARC analysis session creation failed'
message = textwrap.dedent(
f'''\
f"""\
Hello {self.name},
Your analysis session for jSparc could not be created.
Perhaps there was no data for the date and/or stations you selected.
Please try selecting a different cluster or date.
Greetings,
The HiSPARC Team'''
The HiSPARC Team""",
)
sender = 'Beheer HiSPARC <[email protected]>'
send_mail(subject, message, sender, [self.email], fail_silently=False)
12 changes: 6 additions & 6 deletions publicdb/analysissessions/urls.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,10 +4,10 @@

app_name = 'sessions'
urlpatterns = [
path('<slug:slug>/data/', views.data_display, name="data_display"),
path('request/', views.request_form, name="request"),
path('request/validate/', views.validate_request_form, name="validate"),
path('request/<slug:url>/', views.confirm_request, name="confirm"),
path('get_coincidence/', views.get_coincidence, name="get_coincidence"),
path('result/', views.result, name="result"),
path('<slug:slug>/data/', views.data_display, name='data_display'),
path('request/', views.request_form, name='request'),
path('request/validate/', views.validate_request_form, name='validate'),
path('request/<slug:url>/', views.confirm_request, name='confirm'),
path('get_coincidence/', views.get_coincidence, name='get_coincidence'),
path('result/', views.result, name='result'),
]
56 changes: 28 additions & 28 deletions publicdb/analysissessions/views.py
Original file line number Diff line number Diff line change
Expand Up @@ -45,7 +45,7 @@ def get_coincidence(request):
raise
else:
if not session.in_progress():
return error_json(404, "The requested session has not started yet or is already expired.")
return error_json(404, 'The requested session has not started yet or is already expired.')

if not student_name:
student = Student.objects.get(session=session, name='Test student')
Expand All @@ -61,7 +61,7 @@ def get_coincidence(request):
analyzed_coincidence.student = student
analyzed_coincidence.save()
except IndexError:
return error_json(404, "No unanalysed coincidences available, request a new session.")
return error_json(404, 'No unanalysed coincidences available, request a new session.')

events = get_events(analyzed_coincidence)
response = data_json(analyzed_coincidence, events)
Expand All @@ -82,41 +82,41 @@ def get_events(analyzed_coincidence):
continue

timestamp = datetime_to_gps(datetime.combine(event.date, event.time))
event_dict = dict(
timestamp=timestamp,
nanoseconds=event.nanoseconds,
number=event.station.number,
latitude=config.gps_latitude,
longitude=config.gps_longitude,
altitude=config.gps_altitude,
status='on',
detectors=len(event.traces),
traces=event.traces,
pulseheights=event.pulseheights,
integrals=event.integrals,
mips=[ph / 200.0 if ph > 0 else ph for ph in event.pulseheights],
)
event_dict = {
'timestamp': timestamp,
'nanoseconds': event.nanoseconds,
'number': event.station.number,
'latitude': config.gps_latitude,
'longitude': config.gps_longitude,
'altitude': config.gps_altitude,
'status': 'on',
'detectors': len(event.traces),
'traces': event.traces,
'pulseheights': event.pulseheights,
'integrals': event.integrals,
'mips': [ph / 200.0 if ph > 0 else ph for ph in event.pulseheights],
}
events.append(event_dict)
return events


def data_json(coincidence, events):
"""Construct json with data for jSparc to display"""
timestamp = datetime_to_gps(datetime.combine(coincidence.coincidence.date, coincidence.coincidence.time))
data = dict(
pk=coincidence.pk,
timestamp=timestamp,
nanoseconds=coincidence.coincidence.nanoseconds,
events=events,
)
data = {
'pk': coincidence.pk,
'timestamp': timestamp,
'nanoseconds': coincidence.coincidence.nanoseconds,
'events': events,
}
response = HttpResponse(json.dumps(data), content_type='application/json')
response['Access-Control-Allow-Origin'] = '*'
return response


def error_json(error_code, message):
"""Construct error response json for jSparc requests"""
data = dict(message=message, code=error_code)
data = {'message': message, 'code': error_code}
response = HttpResponse(json.dumps(data), status=error_code, content_type='application/json')
response['Access-Control-Allow-Origin'] = '*'
return response
Expand All @@ -143,7 +143,7 @@ def top_lijst(slug):
'wgh_error': wgh_error,
'min_error': min_error,
'num_events': num_events,
}
},
)

return sorted(scores, key=operator.itemgetter('wgh_error'))
Expand Down Expand Up @@ -192,17 +192,17 @@ def result(request):
rank = [x['name'] for x in ranking].index(student_name) + 1
except ValueError:
rank = None
msg = "OK [result stored]"
response = HttpResponse(json.dumps(dict(msg=msg, rank=rank)), content_type='application/json')
msg = 'OK [result stored]'
response = HttpResponse(json.dumps({'msg': msg, 'rank': rank}), content_type='application/json')
response['Access-Control-Allow-Origin'] = '*'
return response


def test_result():
"""Generate random ranking for test sessions"""
msg = "Test session, result not stored"
msg = 'Test session, result not stored'
rank = randint(1, 10)
response = HttpResponse(json.dumps(dict(msg=msg, rank=rank)), content_type='application/json')
response = HttpResponse(json.dumps({'msg': msg, 'rank': rank}), content_type='application/json')
response['Access-Control-Allow-Origin'] = '*'
return response

Expand Down
Loading

0 comments on commit 7835393

Please sign in to comment.