1 changed files with 75 additions and 0 deletions
@ -0,0 +1,75 @@ |
|||||||
|
from util import print_error |
||||||
|
|
||||||
|
try: |
||||||
|
import zbar |
||||||
|
except ImportError: |
||||||
|
print_error("Install zbar package to enable QR scans") |
||||||
|
zbar = None |
||||||
|
|
||||||
|
from urlparse import urlparse, parse_qs |
||||||
|
|
||||||
|
def is_available(): |
||||||
|
if not zbar: |
||||||
|
return False |
||||||
|
return True |
||||||
|
|
||||||
|
def scan_qr(): |
||||||
|
proc = zbar.Processor() |
||||||
|
proc.init() |
||||||
|
proc.visible = True |
||||||
|
|
||||||
|
while True: |
||||||
|
try: |
||||||
|
proc.process_one() |
||||||
|
except: |
||||||
|
# User closed the preview window |
||||||
|
return {} |
||||||
|
|
||||||
|
for r in proc.results: |
||||||
|
if str(r.type) != 'QRCODE': |
||||||
|
continue |
||||||
|
|
||||||
|
return parse_uri(r.data) |
||||||
|
|
||||||
|
def parse_uri(uri): |
||||||
|
if ':' not in uri: |
||||||
|
# It's just an address (not BIP21) |
||||||
|
return {'address': uri} |
||||||
|
|
||||||
|
if '//' not in uri: |
||||||
|
# Workaround for urlparse, it don't handle bitcoin: URI properly |
||||||
|
uri = uri.replace(':', '://') |
||||||
|
|
||||||
|
uri = urlparse(uri) |
||||||
|
result = {'address': uri.netloc} |
||||||
|
|
||||||
|
if uri.path.startswith('?'): |
||||||
|
params = parse_qs(uri.path[1:]) |
||||||
|
else: |
||||||
|
params = parse_qs(uri.path) |
||||||
|
|
||||||
|
for k,v in params.items(): |
||||||
|
if k in ('amount', 'label', 'message'): |
||||||
|
result[k] = v[0] |
||||||
|
|
||||||
|
return result |
||||||
|
|
||||||
|
if __name__ == '__main__': |
||||||
|
# Run some tests |
||||||
|
|
||||||
|
assert(parse_uri('1Marek48fwU7mugmSe186do2QpUkBnpzSN') == |
||||||
|
{'address': '1Marek48fwU7mugmSe186do2QpUkBnpzSN'}) |
||||||
|
|
||||||
|
assert(parse_uri('bitcoin://1Marek48fwU7mugmSe186do2QpUkBnpzSN') == |
||||||
|
{'address': '1Marek48fwU7mugmSe186do2QpUkBnpzSN'}) |
||||||
|
|
||||||
|
assert(parse_uri('bitcoin:1Marek48fwU7mugmSe186do2QpUkBnpzSN') == |
||||||
|
{'address': '1Marek48fwU7mugmSe186do2QpUkBnpzSN'}) |
||||||
|
|
||||||
|
assert(parse_uri('bitcoin:1Marek48fwU7mugmSe186do2QpUkBnpzSN?amount=10') == |
||||||
|
{'amount': '10', 'address': '1Marek48fwU7mugmSe186do2QpUkBnpzSN'}) |
||||||
|
|
||||||
|
assert(parse_uri('bitcoin:1Marek48fwU7mugmSe186do2QpUkBnpzSN?amount=10&label=slush&message=Small%20tip%20to%20slush') == |
||||||
|
{'amount': '10', 'label': 'slush', 'message': 'Small tip to slush', 'address': '1Marek48fwU7mugmSe186do2QpUkBnpzSN'}) |
||||||
|
|
||||||
|
|
||||||
Loading…
Reference in new issue