Jak ładnie obsługiwać KeyboardInterrupt (Ctrl-c) z pycurl?

Question

Piszę skrypt w języku Python, używając pycurl do konsumpcji Twitter's Sreaming API. Oto krótki fragment, który robi dokładnie to, że (wystarczy umieścić swoją Twitter login/hasło, aby go przetestować):

import pycurl 

user = 'USER' 
password = 'PWD' 

def handleData(data): 
    print(data) 

conn = pycurl.Curl() 
conn.setopt(pycurl.USERPWD, "%s:%s" % (user, password)) 
conn.setopt(pycurl.URL, 'https://stream.twitter.com/1/statuses/sample.json') 
conn.setopt(pycurl.WRITEFUNCTION, handleData) 
conn.perform() 

Problemem jest to, że ponieważ skrypt pobiera strumień, conn.perform() nigdy nie powraca (lub bardzo rzadko). Dlatego czasami muszę przerwać skrypt, a KeyboardInterrupt zostanie przechwycony przez metodę perform().

Jednak nie obsługuje go dobrze, drukuje brzydki błąd i podnosi inny wyjątek.

^CTraceback (most recent call last): 
    File "test.py", line 6, in handleData 
    def handleData(data): 
KeyboardInterrupt 
Traceback (most recent call last): 
    File "test.py", line 12, in <module> 
    conn.perform() 
pycurl.error: (23, 'Failed writing body (0 != 2203)') 

cURL FAQ mówi, że aby przerwać trwającą transferu, jedną z funkcji zwrotnych (w moim przypadku handleData) powinny zwrócić szczególną wartość. To świetnie, ale KeyboardInterrupt nie jest przechwytywany przez żadną z funkcji wywołania zwrotnego!

Jak mogę to zrobić starannie?

EDIT: Wiem, że można złapać wyjątki, ale pycurl nadal robi śmieszne rzeczy:

Jeśli zrobić:

try: 
    conn.perform() 
except BaseException as e: 
    print('We caught the exception') 
    print(type(e)) 

uzyskać:

^CTraceback (most recent call last): 
    File "test.py", line 6, in handleData 
    def handleData(data): 
KeyboardInterrupt 
We caught the exception 
<class 'pycurl.error'> 

Oznacza to, że wewnętrznie, pycurl robi jakiś chwyt, drukuje brzydki komunikat o błędzie, a następnie podnosi pycurl.error.

Answer 1

Możesz to zrobić, łapiąc typ pycurl.error. Ex:

try: 
    conn.perform() 
except pycurl.error, e: 
    errorCode, errorText = e.args 
    print 'We got an error. Code: %s, Text:%s'%(errorCode, errorText) 

Answer 2

trzeba złapać CTRL + C i proces, który sygnalizował
oryginalny: Example 1
oryginalny: Example 2

---

Przykład 1

#!/usr/bin/env python 
import signal 
import sys 
def signal_handler(signal, frame): 
     print 'You pressed Ctrl+C!' 
     sys.exit(0) 
signal.signal(signal.SIGINT, signal_handler) 
print 'Press Ctrl+C' 
signal.pause() 

---

Przykład 2

import signal, os 

def handler(signum, frame): 
    print 'Signal handler called with signal', signum 
    raise IOError("Couldn't open device!") 

# Set the signal handler and a 5-second alarm 
signal.signal(signal.SIGALRM, handler) 
signal.alarm(5) 

# This open() may hang indefinitely 
fd = os.open('/dev/ttyS0', os.O_RDWR) 

signal.alarm(0)   # Disable the alarm 

---

A przynajmniej coś nie działa na ten link twitter, zobacz here

• nie zapomnij umieścić Conn.Close (po)conn.perform()

Przydaje się włączenie trybu debugowania podczas testowania.

import pycurl 

username = 'your_user_name' 
password = 'your_password' 

def body(buf): 
    for item in buf.strip().split('\n'): 
     if item.strip(): 
      print item 

def test(debug_type, debug_msg): 
    if len(debug_msg) < 300: 
     print "debug(%d): %s" % (debug_type, debug_msg.strip()) 

conn = pycurl.Curl() 
conn.setopt(pycurl.USERNAME, username) 
conn.setopt(pycurl.PASSWORD, password) 
#conn.setopt(pycurl.SSL_VERIFYPEER, False) 
conn.setopt(pycurl.FOLLOWLOCATION, True) 
conn.setopt(pycurl.VERBOSE, True) 
conn.setopt(pycurl.URL, 'https://stream.twitter.com/1.1/statuses/sample.json') 
conn.setopt(pycurl.DEBUGFUNCTION, test) 
conn.setopt(pycurl.WRITEFUNCTION, body) 
conn.perform() 
conn.close() 

---

Wystarczy skopiować/wklej działa Przykład testowy

➜ ~ hcat twitter.py 
import pycurl 
import signal 
import sys 
from time import sleep 

username = 'bubudee' 
password = 'deebubu' 

def body(buf): 
    for item in buf.strip().split('\n'): 
     if item.strip(): 
      print item 

def test(debug_type, debug_msg): 
    if len(debug_msg) < 300: 
     print "debug(%d): %s" % (debug_type, debug_msg.strip()) 

def handle_ctrl_c(signal, frame): 
    print "Got ctrl+c, going down!" 
    sys.exit(0) 
signal.signal(signal.SIGINT, handle_ctrl_c) 

conn = pycurl.Curl() 
conn.setopt(pycurl.USERNAME, username) 
conn.setopt(pycurl.PASSWORD, password) 
#conn.setopt(pycurl.SSL_VERIFYPEER, False) 
conn.setopt(pycurl.FOLLOWLOCATION, True) 
conn.setopt(pycurl.VERBOSE, True) 
conn.setopt(pycurl.URL, 'https://stream.twitter.com/1.1/statuses/sample.json') 
conn.setopt(pycurl.DEBUGFUNCTION, test) 
conn.setopt(pycurl.WRITEFUNCTION, body) 

conn.perform() 

print "Who let the dogs out?:p" 
sleep(10) 

conn.close() 

➜ ~ python twitter.py 
debug(0): About to connect() to stream.twitter.com port 443 (#0) 
debug(0): Trying 199.16.156.110... 
debug(0): Connected to stream.twitter.com (199.16.156.110) port 443 (#0) 
debug(0): Initializing NSS with certpath: sql:/etc/pki/nssdb 
debug(0): CAfile: /etc/pki/tls/certs/ca-bundle.crt 
    CApath: none 
debug(0): SSL connection using SSL_RSA_WITH_RC4_128_SHA 
debug(0): Server certificate: 
debug(0): subject: CN=stream.twitter.com,OU=Twitter Security,O="Twitter, Inc.",L=San Francisco,ST=California,C=US 
debug(0): start date: Oct 09 00:00:00 2013 GMT 
debug(0): expire date: Dec 30 23:59:59 2016 GMT 
debug(0): common name: stream.twitter.com 
debug(0): issuer: CN=VeriSign Class 3 Secure Server CA - G3,OU=Terms of use at https://www.verisign.com/rpa (c)10,OU=VeriSign Trust Network,O="VeriSign, Inc.",C=US 
debug(0): Server auth using Basic with user 'bubudee' 
debug(2): GET /1.1/statuses/sample.json HTTP/1.1 
Authorization: Basic YnVidWRlZTpkZWVidWJ1 
User-Agent: PycURL/7.29.0 
Host: stream.twitter.com 
Accept: */* 
debug(1): HTTP/1.1 401 Unauthorized 
debug(0): Authentication problem. Ignoring this. 
debug(1): WWW-Authenticate: Basic realm="Firehose" 
debug(1): Content-Type: text/html 
debug(1): Cache-Control: must-revalidate,no-cache,no-store 
debug(1): Content-Length: 1243 
debug(1): Connection: close 
debug(1): 
<html> 
<head> 
<meta http-equiv="Content-Type" content="text/html; charset=utf-8"/> 
<title>Error 401 Unauthorized</title> 
</head> 
<body> 
<h2>HTTP ERROR: 401</h2> 
<p>Problem accessing '/1.1/statuses/sample.json'. Reason: 
<pre> Unauthorized</pre> 
</body> 
</html> 
debug(0): Closing connection 0 
Who let the dogs out?:p 
^CGot ctrl+c, going down!