Question

I have a function like this:

import csv 

def total():
    with open('test.csv', 'r') as csvfile:
        spamreader = csv.reader(csvfile)
        a = [row for row in spamreader]
        return a 
print total()

The output of is:

[['[e-mail]', '1234'], ['[e-mail]', 'metal'], ['[e-mail]', 'hello']]

I want to loop over each username and password until the end of list. The purpose is I want to pass the username and password one by one.
First I want to use [e-mail] and 1234 and then proceed to another.

I tried using:

    def pass_username():
        with open('test.csv', 'r') as csvfile:
            spamreader = csv.reader(csvfile, delimiter=',')
            a = [row[0] for row in spamreader]
            return a 

def pass_password():
    with open('test.csv', 'r') as csvfile:
        spamreader = csv.reader(csvfile, delimiter=',')
        a = [row[1] for row in spamreader]
        return a     

But this returns username at once and password at once like this:

[e-mail]
[e-mail]
[e-mail]
1234
metal
hello

How can I fix this? Thanks

Answer 1

Change this to a single function :

def pass_uname_passwd :
    with open('test.csv', 'r') as csvfile:
                spamreader = csv.reader(csvfile, delimiter=',')
                a = [(row[0], row[1]) for row in spamreader] #Make it a list of 2d tuples
                return a

This will return a list of 2d tuples containing username, password.

list = pass_uname_passwd()

for i in list :
#Use as desired, processing each (uname,passwd) at a time

You can simply do this from your first function as well, by using a for loop after that. The only change i suggest is using tuples instead of lists, to make the data immutable.