2013-12-17 13 views
11

Pobrałem arkusz kalkulacyjny google jako obiekt w pythonie.Używanie openpyxl do odczytu pliku z pamięci

Jak używać openpyxl do korzystania ze skoroszytu bez wcześniejszego zapisania na dysku?

Wiem, że można to zrobić przez xlrd:

book = xlrd.open_workbook(file_contents=downloaded_spreadsheet.read()) 

z „downloaded_spreadsheet” jest moim pobrać plik xlsx jako przedmiot.

Zamiast xlrd, chcę użyć openpyxl z powodu lepszej obsługi xlsx (czytałem).

Używam tego do tej pory ...

#!/usr/bin/python 

    import openpyxl 
    import xlrd 
    # which to use..? 


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="xls"): 

     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__": 



    email = "[email protected]" # (your email here) 
    password = '.....' 
    spreadsheet_id = "......" # (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 
    downloaded_spreadsheet = gs.download(ss) 


    # book = xlrd.open_workbook(file_contents=downloaded_spreadsheet.read(), formatting_info=True) 

    #It works.. alas xlrd doesn't support the xlsx-funcionality that i want... 
    #i.e. being able to read the cell-colordata.. 

Mam nadzieję, że ktoś może pomóc, bo jestem zmaga się od miesięcy, aby uzyskać dane z koloru danej komórki w arkuszu Google. (Wiem, że google-API go nie obsługuje ..)

Odpowiedz

21

W docs dla load_workbook mówi:

#:param filename: the path to open or a file-like object 

..so było zdolne do niego cały czas. Czyta ścieżkę lub bierze obiekt podobny do pliku. miałem tylko przekonwertować mój obiekt plikopodobnym zwrócony przez urlopen, do bytestream z:

from io import BytesIO 
wb = load_workbook(filename=BytesIO(input_excel.read())) 

i mogę czytać każdy kawałek danych w moim Google-kalkulacyjnego.

+0

+1 - Podobny błąd. Czytałem tylko pierwszą połowę i myślałem, że może tylko czytać pliki. Teraz wróciłem i przeczytałem go całkowicie i zobaczyłem, że potrafi on również tworzyć obiekty podobne do plików. –

1

Właściwie tyle jest:

file = open('path/to/file.xlsx', 'rb') 
wb = openpyxl.load_workbook(filename=file) 

i będzie działać. Nie ma potrzeby BytesIO i innych rzeczy.

+1

Nie jest odczytywany z systemu plików, jak wskazuje pytanie. To strumień. –

Powiązane problemy