这段代码有什么问题?它可以很好地下载CSV,但将所有数据放入一列。如果我没有([])括号,它会将每个字符分成单独的单元格。任何想法如何解决这一问题?
import re, urllib, urllib2
class Spreadsheet(object):
def __init__(self, key):
super(Spreadsheet, self).__init__()
self.key = key
class Client(object):
def __init__(self, email, password):
super(Client, self).__init__()
self.email = email
self.password = password
def _get_auth_token(self, email, password, source, service):
url = "https://www.google.com/accounts/ClientLogin"
params = {
"Email": email, "Passwd": password,
"service": service,
"accountType": "HOSTED_OR_GOOGLE",
"source": source
}
req = urllib2.Request(url, urllib.urlencode(params))
return re.findall(r"Auth=(.*)", urllib2.urlopen(req).read())[0]
def get_auth_token(self):
source = type(self).__name__
return self._get_auth_token(self.email, self.password, source, service="wise")
def download(self, spreadsheet, gid=0, format="csv"):
url_format = "https://spreadsheets.google.com/feeds/download/spreadsheets/Export?key=%s&exportFormat=%s&gid=%i"
headers = {
"Authorization": "GoogleLogin auth=" + self.get_auth_token(),
"GData-Version": "3.0"
}
req = urllib2.Request(url_format % (spreadsheet.key, format, gid), headers=headers)
return urllib2.urlopen(req)
if __name__ == "__main__":
import getpass
import csv
email = "username@gmail.com" # (your email here)
password = getpass.getpass()
spreadsheet_id = "18ZzsXwLtwKmm6-nUDXjG4Zknszt8SEQHPmzii_Sz9zY" # (spreadsheet id here)
# Create client and spreadsheet objects
gs = Client(email, password)
ss = Spreadsheet(spreadsheet_id)
# Request a file-like object containing the spreadsheet's contents
csv_file = gs.download(ss)
writer = csv.writer(open(r"C:\Users\me\Desktop\csv.csv", 'wb'))
for row in csv_file:
writer.writerow([row])
这是我的输出:
答案 0 :(得分:0)
csv_file目前只是一个长度为1项的列表,因此行的循环只进行一次。
首先读入csv_file的每一行,然后将其写入别处。
import csv
with open(csv_file, 'rb') as f:
reader = csv.reader(f)
for row in reader:
writer.writerow([row])