def handle_uploaded_file(file, user_id):
user_object = User.objects.get(pk=user_id)
csv_reader = csv.DictReader(file, fieldnames=['date','start','end','duration','project','slip'])
line_data = []
for line in csv_reader:
line_data.append(line)
total_time = 0
val = {}
pickles = {'projects': [], 'slips': [], 'slices': []}
for line in line_data[1:]:
begin = datetime.strptime('T'.join((line['date'], line['start'])), '%Y-%m-%dT%H:%M')
end = datetime.strptime('T'.join((line['date'], line['end'])), '%Y-%m-%dT%H:%M')
total_time += int(line['duration'])
if not val.has_key(line['project']):
project_set = Project.objects.filter(name=line['project'], members = user_id)
if project_set:
if len(project_set) >1:
project = Project()
project.name = line['project']
project_created_msg = _('Found more than one project, you are on, with same name. New project will be created.')
project_created_bool = True
else:
project = project_set[0]
project_created_msg = _('Found project, and will add slips and timeslices to the existing project.')
project_created_bool = False
else:
project_set = Project.objects.filter(name=line['project'])
if project_set:
if len(project_set) >1:
project = Project()
project.name = line['project']
project_created_msg = _('Found more than one project, with same name. New project will be created.')
project_created_bool = True
else:
project = project_set[0]
project_created_msg = _('Found project, and will add user profile, slips and timeslices to the existing project.')
project_created_bool = False
else:
project = Project()
project.name = line['project']
project_created_msg = _('No project of that name. New project will be created')
project_created_bool = True
val[line['project']] = {'created': project_created_bool, 'message': project_created_msg, 'project_object': project, 'slips': {}}
pickles['projects'].append(project)
if not val[line['project']]['slips'].has_key(line['slip']):
if val[line['project']]['created']:
slip = Slip()
slip.name = line['slip']
slip.user = user_object
slip.project = val[line['project']]['project_object']
slip.created = line['date']
slip_created_bool = True
else:
slip_set = Slip.objects.filter(name = line['slip'], user = user_object , project = project)
if not slip_set or len(slip_set) > 1:
slip = Slip()
slip.name = line['slip']
slip.user = user_object
slip.project = val[line['project']]['project_object']
slip.created = line['date']
slip_created_bool = True
else:
slip = slip_set[0]
slip_created_bool = False
val[line['project']]['slips'][line['slip']]={'created': slip_created_bool, 'slip_object': slip, 'slices': []}
pickles['slips'].append(slip)
if val[line['project']]['slips'][line['slip']]['created']:
slice = TimeSlice()
slice.begin = begin
slice.end = end
slice.duration = int(line['duration'])
slice.slip = val[line['project']]['slips'][line['slip']]['slip_object']
slice.user = user_object
slice_created_bool = True
else:
slice_set = TimeSlice.objects.filter(begin = begin, end = end, duration = int(line['duration']), slip = slip, user = user_object)
if slice_set:
if len(slice_set) > 1:
pass
slice = slice_set[0]
slice_created_bool = False
else:
slice = TimeSlice()
slice.begin = begin
slice.end = end
slice.duration = int(line['duration'])
slice.slip = val[line['project']]['slips'][line['slip']]['slip_object']
slice.user = user_object
slice_created_bool = True
val[line['project']]['slips'][line['slip']]['slices'].append(slice)
pickles['slices'].append(slice)
total_time = str(int(total_time/3600.0))+'h'
# Okay, now we have processed the data, lets write it to a couple of files.
import_data = DataImport.objects.create(user=user_object)
file_name = hashlib.sha1(user_object.username + str(time.time())).hexdigest()
import_data.complete_data.save(file_name, ContentFile(pickle.dumps(pickles)))
import_data.partial_data.save(file_name, ContentFile(pickle.dumps(line_data[1:11])))
import_data.save()
return import_data.id
def pinax_handle_uploaded_file(file, user_id):
user_object = User.objects.get(pk=user_id)
csv_reader = csv.DictReader(file, fieldnames=['date','start','end','duration','project','task'])
line_data = []
for line in csv_reader:
line_data.append(line)
total_time = 0
val = {}
pickles = {'projects': [], 'tasks': [], 'slices': []}
for line in line_data[1:]:
begin = datetime.strptime('T'.join((line['date'], line['start'])), '%Y-%m-%dT%H:%M')
end = datetime.strptime('T'.join((line['date'], line['end'])), '%Y-%m-%dT%H:%M')
total_time += int(line['duration'])
if not val.has_key(line['project']):
project_set = Project.objects.filter(name=line['project'], member_users=user_id)
if project_set:
if len(project_set) > 1:
project, project_created_msg, project_created_bool = create_pinax_project(user_object, line)
else:
project = project_set[0]
project_created_msg = _('Found project, and will add tasks and timeslices to the existing project.')
project_created_bool = False
else:
project_set = Project.objects.filter(name=line['project'])
if project_set:
if len(project_set) >1:
project, project_created_msg, project_created_bool = create_pinax_project(user_object, line)
else:
project = project_set[0]
project_created_msg = _('Found project will add tasks and timeslices to the existing project.')
project_created_bool = False
else:
project, project_created_msg, project_created_bool = create_pinax_project(user_object, line)
val[line['project']] = {'created': project_created_bool, 'message': project_created_msg, 'project_object': project, 'tasks': {}}
pickles['projects'].append(project)
if not val[line['project']]['tasks'].has_key(line['task']):
if val[line['project']]['created']:
task = Task()
task.summary = line['task']
task.creator = user_object
task.group = val[line['project']]['project_object']
task.created = line['date']
task_created_bool = True
else:
task_set = Task.objects.filter(summary=line['task'], creator=user_object, object_id=project.id)
if not task_set or len(task_set) > 1:
task = Task()
task.summary = line['task']
task.creator = user_object
task.group = val[line['project']]['project_object']
task.created = line['date']
task_created_bool = True
else:
task = task_set[0]
task_created_bool = False
val[line['project']]['tasks'][line['task']]={'created': task_created_bool, 'task_object': task, 'slices': []}
pickles['tasks'].append(task)
if val[line['project']]['tasks'][line['task']]['created']:
tslice = TimeSlice()
tslice.begin = begin
tslice.duration = int(line['duration'])
tslice.task = val[line['project']]['tasks'][line['task']]['task_object']
tslice.user = user_object
slice_created_bool = True
else:
slice_set = TimeSlice.objects.filter(begin=begin, duration=int(line['duration']), task=task, user=user_object)
if slice_set:
tslice = slice_set[0]
slice_created_bool = False
else:
tslice = TimeSlice()
tslice.begin = begin
tslice.duration = int(line['duration'])
tslice.task = val[line['project']]['tasks'][line['task']]['task_object']
tslice.user = user_object
slice_created_bool = True
val[line['project']]['tasks'][line['task']]['slices'].append(tslice)
pickles['slices'].append(tslice)
total_time = str(int(total_time/3600.0))+'h'
# Okay, now we have processed the data, lets write it to a couple of files.
import_data = DataImport.objects.create(user=user_object)
file_name = hashlib.sha1(user_object.username + str(time.time())).hexdigest()
import_data.complete_data.save(file_name, ContentFile(pickle.dumps(pickles)))
import_data.partial_data.save(file_name, ContentFile(pickle.dumps(line_data[1:11])))
import_data.save()
return import_data.id