def index(request):
# If this is a form submission
if request.method == "POST":
import copy
data = copy.copy(request.POST)
# In case we're not in production
# Remove this line before production!
request = views.get_current_user(request)
data['poster'] = str(request.user)
form = CommentForm(data)
if form.is_valid():
form.save(commit=True)
return HttpResponseRedirect(request.META.get('HTTP_REFERER'))
data['giver'] = str(request.user)
form = StarForm(data)
if form.is_valid():
form.save(commit=True)
return HttpResponseRedirect(request.META.get('HTTP_REFERER'))
form = EmployeeForm(request.POST)
if form.is_valid():
form.save(commit=True)
return HttpResponseRedirect(request.META.get('HTTP_REFERER'))
return views.load_page(request, 'roster/index.html', prep_context())
def filter_near_shifts(request):
request_user = get_current_user(request)
user = Employees.objects.get(netid=request_user.user)
# Get the shift of the given id
shift_id = request.GET.get('shiftID', None)
this_shift = Shifts.objects.get(event_id=shift_id)
# First we filter all shifts of the same location
filtered_shifts = Shifts.objects.filter(location=this_shift.location)
# Then we filter based on shifts that are occuring directly before or after the given shift
filtered_shifts = filtered_shifts.filter(Q(shift_end=this_shift.shift_start) | Q(shift_start=this_shift.shift_end)).order_by('shift_start')
# If the count is 2 then we should have exactly 2 shifts (one before and one after)
# This allows us to create a range from the shift before to the shift after
if filtered_shifts.count() == 2:
start = filtered_shifts.first().shift_start
end = filtered_shifts.last().shift_end
print(start)
print(end)
elif filtered_shifts.count() == 1:
start = filtered_shifts.first().shift_start
end = this_shift.shift_end
print(start)
print(end)
else:
start = this_shift.shift_start
end = this_shift.shift_end
print(start)
print(end)
# If this is an open shift, we also need the information from the corresponding shift_cover entry
if this_shift.is_open:
translated_shift_cover = {
'poster': str(this_shift.owner),
'sobstory': this_shift.sob_story if this_shift.sob_story else "No sob story recorded",
}
else:
translated_shift_cover = ['Not an Open Shift']
# We construct a dictonary of the values contained in this shift
translated_this_shift = {
'event_id': this_shift.event_id,
'title': this_shift.title,
'owner': str(this_shift.owner),
'shift_date': this_shift.shift_date,
'shift_start': this_shift.shift_start,
'shift_end': this_shift.shift_end,
'location': this_shift.location,
'is_open': this_shift.is_open,
'checked_in': this_shift.checked_in,
'permanent_id': this_shift.permanent_id
}
translated_shifts = filtered_shifts.values('event_id', 'title', 'owner', 'shift_date', 'shift_start', 'shift_end', 'location', 'is_open','checked_in', 'permanent_id')
data = {
'shiftCover': translated_shift_cover,
'thisShift': translated_this_shift,
'shifts': list(translated_shifts)
}
return JsonResponse(data)
def filter_open_shifts(request):
request_user = get_current_user(request)
user = Employees.objects.get(netid=request_user.user)
option = request.GET.get('option', None) # Retreive the option
date_string = request.GET.get('date', None) # Retreive the date entered
location = request.GET.get('location', None) # Retreive the desired location of open shifts
date = datetime.strptime(date_string, '%Y-%m-%d') # Parse that string into a datetime object
queryset = Shifts.objects.filter(is_open = True) # Retrieve the shifts that are open
# If the location is a simple location
if location in ['ma', 'da', 'st', 'rc', 'sd', 'ss']:
queryset = queryset.filter(location = location).order_by('shift_start') # We filter the query set down to the shifts in that location
# Otherwise, we need to determine by position, and filter with the corresponding locations
else:
if location in ['lbt', 'llt']: # If they are a labtech or lead labtech we need to show all library shifts
queryset = queryset.filter(Q(location='ma') | Q(location='da') | Q(location='st')).order_by('shift_start')
elif location == 'spt': # If support tech
queryset = queryset.filter(Q(location='rc') | Q(location='sd')).order_by('shift_start')
elif location == 'sst': # If senior support tech
queryset = queryset.filter(location='ss')
elif location == 'sdr': # If support desk rep
queryset = queryset.filter(location='sr').order_by('shift_start')
elif location == 'mgr': # If manager
queryset = queryset.filter(location='md').order_by('shift_start')
else: # This shouldn't be possible, but in this case I am showing all shifts
queryset = queryset.order_by('shift_start')
# Contextual options
if option == 'next':
date = date + timedelta(days=7)
elif option == 'prev':
date = date - timedelta(days=7)
# User shifts explains what is happening here pretty well
given_week_day_iso = date.isocalendar()[2]
# if currently is sunday
if given_week_day_iso == 7:
sunday_start_date_current_week = date
else:
sunday_start_date_current_week = date - timedelta(days=given_week_day_iso)
end_of_week = sunday_start_date_current_week + timedelta(days=6)
filtered_shifts = queryset.filter(shift_date__gte=sunday_start_date_current_week, shift_date__lte=end_of_week)
translated_shifts = filtered_shifts.values('event_id', 'title', 'owner', 'shift_date', 'shift_start', 'shift_end', 'location', 'is_open', 'checked_in', 'permanent_id')
week = [] # start empty
for day in range(0, 7):
week.append(sunday_start_date_current_week + timedelta(days=day))
data = {
'date': str(date),
'shifts': list(translated_shifts),
'week': week
}
return JsonResponse(data)
def index(request):
# If this is a form submission
if request.method == "POST":
# Need to validate which form is being submitted (give the input a name attribute)
import copy
data = copy.copy(request.POST)
form = None
if 'e-form' in request.POST:
# In case we're not in production
# Remove this line before production!
request = views.get_current_user(request)
data['announcer'] = str(request.user)
form = EventForm(data)
print(form.fields)
if 'a-form' in request.POST:
# In case we're not in production
# Remove this line before production!
request = views.get_current_user(request)
data['announcer'] = str(request.user)
form = AnnouncmentForm(data)
print(form)
if 's-form' in request.POST:
status = StaffStatus.objects.get(netid=request.POST['netid'])
form = StatusForm(data, instance=status)
print(form.fields)
if form is not None and form.is_valid():
form.save(commit=True)
return HttpResponseRedirect(request.META.get('HTTP_REFERER'))
else:
print("Bad form!")
return views.load_page(request, 'homebase/index.html', prep_context())
def post_discipline(request):
# Make sure it's a post request
if not request.method == 'POST':
return HttpResponse(json.dumps({"status": "Failed!"}),
content_type="application/json")
# Make sure the user has proper access rights to do this
request = views.get_current_user(request)
poster = str(request.user)
about = request.POST.get('about', None)
if about is not None:
emp = Employees.objects.get(netid=poster)
if emp.position == 'llt':
access_area = 'roster_modfb_lab'
else:
access_area = 'roster_modfb_all'
print(access_area)
else:
return HttpResponse(json.dumps(
{"status": "Failed! About netid not found"}),
content_type="application/json")
if not get_access(poster, access_area):
return HttpResponse(json.dumps(
{"status": "Failed! User does not have access"}),
content_type="application/json")
# Finish getting variables
subject = request.POST.get('subject', None)
body = request.POST.get('body', None)
extent = request.POST.get('extent', None)
# Construct discipline object
comment = Discipline(
subject=subject,
poster=Employees.objects.get(netid=poster),
about=Employees.objects.get(netid=about),
val=extent,
description=body,
)
print(comment)
comment.save()
return HttpResponse(json.dumps({"status":
"Comment successfully created!"}),
content_type="application/json")
def index(request):
# If this is a form submission
if request.method == "POST":
# Copy POST data into a mutable variable
import copy
data = copy.copy(request.POST)
# In case we're not in production
# Remove this line before production!
request = views.get_current_user(request)
data['netid'] = str(request.user)
form = StatusLogForm(data)
if form.is_valid():
form.save(commit=True)
return HttpResponseRedirect(request.META.get('HTTP_REFERER'))
return views.load_page(request, 'printinfo/index.html', prep_context())
def post_award(request):
# Make sure it's a post request
if not request.method == 'POST':
return HttpResponse(json.dumps({"status": "Failed!"}),
content_type="application/json")
# Make sure the user has proper access rights to do this
request = views.get_current_user(request)
giver = str(request.user)
recipient = request.POST.get('recipient', None)
if recipient is not None:
emp = Employees.objects.get(netid=giver)
if emp.position == 'llt':
access_area = 'roster_modfb_lab'
else:
access_area = 'roster_modfb_all'
else:
return HttpResponse(json.dumps(
{"status": "Failed! User does not have access(1)"}),
content_type="application/json")
if not get_access(giver, access_area):
return HttpResponse(json.dumps(
{"status": "Failed! User does not have access(2)"}),
content_type="application/json")
# Finish getting variables
name = request.POST.get('name', None)
reason = request.POST.get('reason', None)
award = request.POST.get('type', None)
# Construct discipline object
award = Trophies(
name=name,
trophy_type=award,
giver=Employees.objects.get(netid=giver),
recipient=Employees.objects.get(netid=recipient),
reason=reason,
)
print(award)
award.save()
return HttpResponse(json.dumps({"status": "Award successfully created!"}),
content_type="application/json")
def filter_user_shifts(request):
request_user = get_current_user(request)
user = Employees.objects.get(netid=request_user.user)
option = request.GET.get('option', None) # Retrieve the option
date_string = request.GET.get('date', None) # Retrieve the date entered
date = datetime.strptime(date_string, '%Y-%m-%d') # Make that string into a datetime object
# We filter the query set down to the shifts owned by user logged in
queryset = Shifts.objects.filter(owner=user).order_by('shift_start')
# The request can be contextual to the previous or next week, this is triggered when the user presses those buttons
if option == 'next':
date = date + timedelta(days=7) # We add 7 days to the current date
elif option == 'prev':
date = date - timedelta(days=7) # We remove 7 days from the current date
# We get the iso date number, unfortunately, ISO believes that the week starts on Monday, so we have to work around
given_week_day_iso = date.isocalendar()[2]
# if currently is Sunday, remember ISO thinks Sunday is the last day of the week
if given_week_day_iso == 7:
sunday_start_date_current_week = date
else:
sunday_start_date_current_week = date - timedelta(days=given_week_day_iso) # To find our Sunday start we subtract the current day of the week via timedelta
end_of_week = sunday_start_date_current_week + timedelta(days=6) # Then we add 6 to find end of this week, which is clearly a Saturday
# next we will generate a "week" which will be a list of dates
week = [] # start empty, we will build up the dates of each day
for day in range(0, 7): # day 1-7
week.append(sunday_start_date_current_week + timedelta(days=day)) # Add each week date, Sunday to Saturday
filtered_shifts = queryset.filter(shift_date__gte=sunday_start_date_current_week, shift_date__lte=end_of_week) # Now we filter the shifts to be the ones in the date range
# We have to translate these shifts into these values so they can be JSON
translated_shifts = filtered_shifts.values('event_id', 'title', 'owner','shift_date', 'shift_start','shift_end', 'location', 'is_open', 'checked_in', 'permanent_id')
# This is the formatted data that we will be returning via AJAX
data = {
'date': str(date),
'shifts': list(translated_shifts),
'week': week
}
# We return the data as JSON
return JsonResponse(data)
def take_cover(request):
request = get_current_user(request)
permanent = request.POST.get('permanent', None) == 'true'
partial = request.POST.get('partial', None) == 'true'
sob_story = str(request.POST.get('sob_story', None))
# actor = Employees.objects.get(netid=str((Shifts.objects.get(event_id=request.GET.get('event_id', None))).owner))
actor = Employees.objects.get(netid=str(request.user))
# actor = get_current_user(request)
if permanent:
s_id = str(request.POST.get('permanent_id', None))
else:
s_id = str(request.POST.get('event_id', None))
if partial:
part_start = request.POST.get('part_start', None)
part_end = request.POST.get('part_end', None)
else:
part_start = None
part_end = None
data = CoverInstructions(
post=False,
permanent=permanent,
partial=partial,
shift_id=s_id,
actor=actor,
start_time=part_start,
end_time=part_end,
sob_story=sob_story,
)
post_status = data.push()
synchronize(flush=False)
sleep(2) # Wait for async fns to finish
json_data = {
'pst_status': post_status
}
# We return the data as JSON
return JsonResponse(json_data)
def index(request):
request_user = get_current_user(request)
user = Employees.objects.get(netid=request_user.user)
user_position = user.position
# The possible positions for an employee in the database
positions = {
'lbt': "Lab Tech",
'llt': "Lead Lab Tech",
'spt': "Support Tech",
'sst': "Senior Support Tech",
'sdr': "Support Desk Rep",
'mgr': "Manager"
}
# The possible locations for a location in the database
locations = {
'ma': "Marshburn",
'da': "Darling",
'st': "Stamps",
'rc': "Repair Center",
'sd': "Support Desk",
'md': "MoD Desk"
}
# We import the current time
now = timezone.now()
# Synchronize our db with the Google Cal
synchronize(flush=False)
# We build our context for the page
context = {
"date": now,
"positions": positions,
"locations": locations,
"user_position": user_position
}
return views.load_page(request, 'shifts/index.html', context)
def post_cover(request):
request = get_current_user(request)
permanent = request.POST.get('permanent', None) == 'true'
partial = request.POST.get('partial', None) == 'true'
sob_story = str(request.POST.get('sob_story', None))
actor = Employees.objects.get(netid=str(request.user))
shift = Shifts.objects.get(event_id=request.POST.get('event_id', None))
shift_owner = shift.owner
# Check for bad user posting data
if shift_owner.netid != str(request.user) or not get_access(str(request.user), "shift_postall"):
# if the user can't post this shift
json_data = {
'shift_owner': shift_owner.netid,
'user': str(request.user),
'pst_status': "Bad user",
}
return JsonResponse(json_data)
if permanent:
s_id = str(request.POST.get('permanent_id', None))
else:
s_id = str(request.POST.get('event_id', None))
if partial:
part_start = request.POST.get('part_start', None)
part_end = request.POST.get('part_end', None)
else:
part_start = None
part_end = None
"""
# Test data
permanent = True
partial = False
actor = Employees.objects.get(netid='nchera13')
sob_story = "tst"
s_id = "2um9rcro6hhk7bb0ttee4debb1"
part_start = None
part_end = None
"""
data = CoverInstructions(
post=True,
permanent=permanent,
partial=partial,
shift_id=s_id,
actor=actor,
start_time=part_start,
end_time=part_end,
sob_story=sob_story,
)
post_status = data.push()
synchronize(flush=False)
sleep(2) # Wait for async fns to finish
json_data = {
'pst_status': post_status
}
# We return the data as JSON
return JsonResponse(json_data)