Adding data to a CSV file through Python

Question

I've got a function that adds new data to a csv file. I've got it somewhat working. However, I'm having a few problems.

When I add the new values (name, phone, address, birthday), it adds them all in one column, rather than separate columns in the same row. (Not really much idea on how to split them up in various columns...)
I can only add numbers rather than string values. So if I write add_friend(blah, 31, 12, 45), it will come back saying blah is not defined. However, if I write add_friend(3,4,5,6), it'll add that to the new row—but, into a single column
An objective with the function is: If you try and add a friend that's already in the csv (say, Bob), and his address, phone, birthday are already in the csv, if you add_friend(Bob, address, phone, birthday), it should state False, and not add it. However, I have no clue how to do this. Any ideas?

Here is my code:

def add_friend (name, phone, address, birthday):
    with open('friends.csv', 'ab') as f:
       newrow = [name, phone, address, birthday]
       friendwriter = csv.writer(open('friends.csv', 'ab'), delimiter=' ',
                                   quotechar='|', quoting=csv.QUOTE_MINIMAL)
      friendwriter.writerow(newrow)
      #friendreader = csv.reader(open('friends.csv', 'rb'), delimiter=' ', quotechar='|')
      #for row in friendreader:
        #print ' '.join(row)

     print newrow

@Steven Seriously, you could have updated your other question, which is really the exact same question. Better to keep it all together so everyone can help you through the progression to solving your problem. I vote to merge. — Keith Layne
– Keith Layne, Commented Aug 31, 2011 at 14:54

score 5 · Accepted Answer · 2011-09-01 03:48:54Z

Based on your requirements, and what you appear to be trying to do, I've written the following. It should be verbose enough to be understandable.

You need to be consistent with your delimiters and other properties when reading the CSV files.

Also, try and move "friends.csv" to a global, or at least in some non-hard coded constant.

import csv

def print_friends():
    reader = csv.reader(open("friends.csv", "rb"), delimiter=' ', quotechar='|', quoting=csv.QUOTE_MINIMAL)
    for row in reader:
        print row

def friend_exists(friend):
    reader = csv.reader(open("friends.csv", "rb"), delimiter=' ', quotechar='|', quoting=csv.QUOTE_MINIMAL)
    for row in reader:
        if (row == friend):
            return True
    return False

def add_friend(name, phone, address, birthday):
    friend = [name, phone, address, birthday]
    if friend_exists(friend):
        return False

    writer = csv.writer(open("friends.csv", "ab"), delimiter=' ', quotechar='|', quoting=csv.QUOTE_MINIMAL)
    writer.writerow(friend)
    return True

print "print_friends: "
print_friends()

print "get_friend: "
test_friend = ["barney", "4321 9876", "New York", "2000"]
print friend_exists(test_friend)

print "add_friend: "
print add_friend("barney", "4321 9876", "New York", "2000")

Devin Jeanpierre · Accepted Answer · 2011-08-31 13:50:16Z

1

It doesn't do that. What makes you think that's what it does? It's possible that the quoting scheme you really want isn't the one you specified to csv.writer: i.e., spaces delimit columns, and | is the quoting character.
blah is not a string literal, "blah" is. blah without quotes is a variable reference, and the variable didn't exist here.
In order to check whether a name is already in the CSV file, you have to read the whole CSV file first, checking for the details. Open the file twice: first for reading ('r'), and use csv.reader to turn that into a row-iterator and find all the names. You can add those names to a set, and then check with that set forever after.

re #3:

To get this set, you could define a function as so:

def get_people():
    with open(..., 'r') as f:
        return set(map(tuple, csv.reader(f)))

And then if you assigned the set somewhere, e.g. existing_people = get_people() you could then check against it when adding new people, as follows:

newrow = (name, phone, address, birthday)
if newrow in existing_people:
    return False
else:
    existing_people.add(newrow)
    friendwriter.writerow(newrow)

answered Aug 31, 2011 at 13:50

Devin Jeanpierre

96.1k5 gold badges59 silver badges80 bronze badges

2 Comments

Stephen Over a year ago

re 1. That is what it does. I have been adding in values to the csv constantly. It just adds number values to the first column. Not sure how to do it for strings still? Thanks for your help on 3!!

Devin Jeanpierre Over a year ago

@Stephen try the following code:

import csv; from StringIO import StringIO; x = StringIO(); csv.writer(x, delimiter=' ', quotechar='|', quoting=csv.QUOTE_MINIMAL).writerow(("a", "b", 2, 3)); print x.getvalue()

-- does that have four columns or one? If you say it has four, then I don't see the problem. If you say it has one, then you picked the wrong column delimiter, because that is four columns under the column delimiter you chose, but only one if you chose a different column delimiter.

Zearin · Accepted Answer · 2013-02-26 15:44:56Z

You aren't stating how experienced with Python you already are, so I am aiming this a little low - no offence intended

There are several "requirements" for your homework. In general, you should ensure that one function does one thing. So, to meet all your requirements, you’ll need several functions; look at creating at least one module (i.e., a file with functions in it).
A space delimiter and a | for quotes is pretty unusual. For the current file, what is the delimieter between columns? And what is used to quote/escape text? (By “escaping text”, I mean: If I have a csv file that uses commas as the column delimiter, and I want to put a sentence with commas into just one column, I need to tell the difference between a comma that means “new column” and a comma that is part of a sentence in a column. Microsoft decided that Excel would support double quotes—so "hello, sailor" became a de facto standard.
If you want to know if "bob brown” is already in the file, you will need to read the whole file first before trying to insert. You can do this using 'r', then 'w'. But should you read the whole file every time you want to insert one record? What if you have a hundred records to add—should you read the whole file each time? Is there a way to store the names during the adding process?
blah is not a string. It needs to be quoted to be a string literal ("blah"). blah just refers to a variable whose name is blah. If it says blah is not defined, that’s because you have not declared the variable blah to hold anything.

Collectives™ on Stack Overflow

Adding data to a CSV file through Python

3 Answers 3

Comments

2 Comments

Comments

Your Answer

Linked

Hot Network Questions

Collectives™ on Stack Overflow

3 Answers 3

Comments

2 Comments

Comments

Your Answer

Sign up or log in

Post as a guest

Linked

Related