Bare start on CSV support
This commit is contained in:
parent
84c441cf39
commit
60cd26d227
2 changed files with 52 additions and 0 deletions
4
todo/data/import example.csv
Normal file
4
todo/data/import example.csv
Normal file
|
@ -0,0 +1,4 @@
|
||||||
|
Title,Group,Task List,Created Date,Due Date,Completed,Created By,Assigned To,Note,Priority
|
||||||
|
Make dinner,Scuba Divers,Example Llist,2012-03-14,,No,shacker,user1,This is as good as it gets,3
|
||||||
|
Bake bread,Scuba Divers,Example Llist,2012-03-14,2012-03-14,,shacker,,,
|
||||||
|
Eat food,Scuba Divers,Example Llist,,2015-06-24,Yes,,user2,Every generation throws a hero up the pop charts,77
|
|
48
todo/management/commands/import_csv.py
Normal file
48
todo/management/commands/import_csv.py
Normal file
|
@ -0,0 +1,48 @@
|
||||||
|
import csv
|
||||||
|
import unicodecsv
|
||||||
|
import sys
|
||||||
|
from pathlib import Path
|
||||||
|
from typing import Any
|
||||||
|
|
||||||
|
from django.core.management.base import BaseCommand, CommandParser
|
||||||
|
|
||||||
|
from django.contrib.auth.models import Group
|
||||||
|
from django.contrib.auth import get_user_model
|
||||||
|
|
||||||
|
from todo.models import Task, TaskList
|
||||||
|
|
||||||
|
|
||||||
|
class Command(BaseCommand):
|
||||||
|
help = """Import specifically formatted CSV file of incoming tasks.
|
||||||
|
For specfic format of inbound CSV, see data/import example.csv.
|
||||||
|
For documentation on field formats and required fields, see README.md.
|
||||||
|
"""
|
||||||
|
|
||||||
|
def add_arguments(self, parser: CommandParser) -> None:
|
||||||
|
|
||||||
|
parser.add_argument(
|
||||||
|
"-f", "--file", dest="file", default=None, help="File to to inbound CSV file."
|
||||||
|
)
|
||||||
|
|
||||||
|
def handle(self, *args: Any, **options: Any) -> None:
|
||||||
|
# ### Sanity checks ###
|
||||||
|
|
||||||
|
# Need a file to proceed
|
||||||
|
if not options.get("file"):
|
||||||
|
print("Sorry, we need a file name to work from.")
|
||||||
|
sys.exit(1)
|
||||||
|
else:
|
||||||
|
print(options.get("file"))
|
||||||
|
if not Path(options.get("file")).exists():
|
||||||
|
print(f"Sorry, couldn't find file name specified: {options.get('file')}")
|
||||||
|
sys.exit(1)
|
||||||
|
|
||||||
|
print("Have arg and good file path")
|
||||||
|
with open(Path(options.get("file")), 'rb') as csvfile:
|
||||||
|
# csvreader = csv.reader(csvfile, delimiter=' ', quotechar='|')
|
||||||
|
# csvreader = csv.DictReader(csvfile)
|
||||||
|
csvreader = unicodecsv.reader(csvfile, encoding='utf-8-sig')
|
||||||
|
for row in csvreader:
|
||||||
|
# print(', '.join(row))
|
||||||
|
# print(row['Title'], row['Group'])
|
||||||
|
print(row)
|
Loading…
Add table
Add a link
Reference in a new issue