Miguel Freitas
11 years ago
2 changed files with 180 additions and 0 deletions
@ -0,0 +1,70 @@
@@ -0,0 +1,70 @@
|
||||
from cgi import escape |
||||
class HTML(object): |
||||
'''Easily generate HTML. |
||||
|
||||
>>> h = HTML() |
||||
>>> p = h.p('hello, world!') |
||||
>>> p.text('more text') |
||||
>>> with h.table(border='1', newlines=True): |
||||
... for i in range(2): |
||||
... with h.tr: |
||||
... h.td('he<l>lo', a='"foo"') |
||||
... h.td('there') |
||||
... |
||||
>>> print h |
||||
<p>hello, world!more text</p> |
||||
<table border="1"> |
||||
<tr><td a=""foo"">he<l>lo</td><td>there</td></tr> |
||||
<tr><td a=""foo"">he<l>lo</td><td>there</td></tr> |
||||
</table> |
||||
|
||||
''' |
||||
def __init__(self, name=None, stack=None): |
||||
self.name = name |
||||
self.content = [] |
||||
self.attrs = {} |
||||
# insert newlines between content? |
||||
self.newlines = False |
||||
if stack is None: |
||||
stack = [self] |
||||
self.stack = stack |
||||
def __getattr__(self, name): |
||||
# adding a new tag or newline |
||||
if name == 'newline': |
||||
e = '\n' |
||||
else: |
||||
e = HTML(name, self.stack) |
||||
self.stack[-1].content.append(e) |
||||
return e |
||||
def text(self, text): |
||||
# adding text |
||||
self.content.append(escape(text)) |
||||
def __call__(self, *content, **kw): |
||||
# customising a tag with content or attributes |
||||
if content: |
||||
self.content = map(escape, content) |
||||
if 'newlines' in kw: |
||||
# special-case to allow control over newlines |
||||
self.newlines = kw.pop('newlines') |
||||
for k in kw: |
||||
self.attrs[k] = escape(kw[k]).replace('"', '"') |
||||
return self |
||||
def __enter__(self): |
||||
# we're now adding tags to me! |
||||
self.stack.append(self) |
||||
return self |
||||
def __exit__(self, exc_type, exc_value, exc_tb): |
||||
# we're done adding tags to me! |
||||
self.stack.pop() |
||||
def __str__(self): |
||||
# turn me and my content into text |
||||
join = '\n' if self.newlines else '' |
||||
if self.name is None: |
||||
return join.join(map(str, self.content)) |
||||
a = ['%s="%s"'%i for i in self.attrs.items()] |
||||
l = [self.name] + a |
||||
s = '<%s>%s'%(' '.join(l), join) |
||||
if self.content: |
||||
s += join.join(map(str, self.content)) |
||||
s += join + '</%s>'%self.name |
||||
return s |
@ -0,0 +1,110 @@
@@ -0,0 +1,110 @@
|
||||
#!/usr/bin/python |
||||
# |
||||
# This sample script is a username crawler: it will obtain all known usernames |
||||
# from block chain and then try to download avatar and profiles for all of |
||||
# them. The report is shown as an html file. |
||||
# |
||||
# Downloaded data is cached in a python pickle file, so it may be executed |
||||
# again and it won't need to get everything all over again (you may run it |
||||
# from cron scripts, for example) |
||||
|
||||
import sys, cPickle, time |
||||
|
||||
dbFileName = "usernameCrawler.pickle" |
||||
htmlFileName = "userlist.html" |
||||
cacheTimeout = 24*3600 |
||||
|
||||
try: |
||||
from bitcoinrpc.authproxy import AuthServiceProxy |
||||
except ImportError as exc: |
||||
sys.stderr.write("Error: install python-bitcoinrpc (https://github.com/jgarzik/python-bitcoinrpc)\n") |
||||
exit(-1) |
||||
|
||||
serverUrl = "http://user:pwd@127.0.0.1:28332" |
||||
if len(sys.argv) > 1: |
||||
serverUrl = sys.argv[1] |
||||
|
||||
twister = AuthServiceProxy(serverUrl) |
||||
|
||||
class User: |
||||
avatar = "" |
||||
fullname = "" |
||||
location = "" |
||||
updateTime = 0 |
||||
|
||||
class MyDb: |
||||
lastBlockHash = 0 |
||||
|
||||
try: |
||||
db = cPickle.load(open(dbFileName)) |
||||
nextHash = db.lastBlockHash |
||||
except: |
||||
db = MyDb() |
||||
db.usernames = {} |
||||
nextHash = twister.getblockhash(0) |
||||
|
||||
while True: |
||||
block = twister.getblock(nextHash) |
||||
db.lastBlockHash = block["hash"] |
||||
print str(block["height"]) + "\r", |
||||
usernames = block["usernames"] |
||||
for u in usernames: |
||||
if not db.usernames.has_key(u): |
||||
db.usernames[u] = User() |
||||
if block.has_key("nextblockhash"): |
||||
nextHash = block["nextblockhash"] |
||||
else: |
||||
break |
||||
|
||||
now = time.time() |
||||
for u in db.usernames.keys(): |
||||
if db.usernames[u].updateTime + cacheTimeout < now: |
||||
|
||||
print "getting avatar for", u, "..." |
||||
d = twister.dhtget(u,"avatar","s") |
||||
if len(d) == 1 and d[0].has_key("p") and d[0]["p"].has_key("v"): |
||||
db.usernames[u].avatar = d[0]["p"]["v"] |
||||
|
||||
print "getting profile for", u, "..." |
||||
d = twister.dhtget(u,"profile","s") |
||||
if len(d) == 1 and d[0].has_key("p") and d[0]["p"].has_key("v"): |
||||
db.usernames[u].fullname = d[0]["p"]["v"]["fullname"] |
||||
db.usernames[u].location = d[0]["p"]["v"]["location"] |
||||
|
||||
db.usernames[u].updateTime = now |
||||
|
||||
cPickle.dump(db,open(dbFileName,"w")) |
||||
|
||||
|
||||
from HTML import HTML |
||||
from cgi import escape |
||||
def outputHtmlUserlist(fname, db, keys): |
||||
h = HTML() |
||||
head = h.head("") |
||||
with h.body(""): |
||||
with h.table(border='1', newlines=True): |
||||
with h.colgroup: |
||||
h.col(span="1", style="width: 64px;") |
||||
h.col(span="1", style="width: 130px;") |
||||
h.col(span="1", style="width: 250px;") |
||||
h.col(span="1", style="width: 250px;") |
||||
with h.tr: |
||||
h.th("avatar") |
||||
h.th("username") |
||||
h.th("fullname") |
||||
h.th("location") |
||||
for u in keys: |
||||
with h.tr: |
||||
with h.td(): |
||||
h.img('',src=escape(db.usernames[u].avatar), width="64", height="64") |
||||
h.td(u) |
||||
h.td(escape(db.usernames[u].fullname)) |
||||
h.td(escape(db.usernames[u].location)) |
||||
open(fname, "w").write(str(h)) |
||||
|
||||
print "Generating", htmlFileName, "..." |
||||
|
||||
keys = db.usernames.keys() |
||||
keys.sort() # sorted by username |
||||
outputHtmlUserlist(htmlFileName, db, keys) |
||||
|
Loading…
Reference in new issue