added get_week.py to handle networking
This commit is contained in:
parent
3045456ed5
commit
9eb80cf402
28
get_week.py
Normal file
28
get_week.py
Normal file
@ -0,0 +1,28 @@
|
||||
# -*- coding: utf-8 -*-
|
||||
from requests import get
|
||||
from lxml import html
|
||||
import re
|
||||
|
||||
|
||||
def get_week():
|
||||
page = get('https://www.kth.bme.hu/')
|
||||
domtree = html.fromstring(page.content)
|
||||
|
||||
text = domtree.cssselect('#site-container > div.infobar > div.line-2')[0].text_content()
|
||||
text = unicode(text.replace('\n', '').replace('\t', ''))
|
||||
|
||||
regex = re.match(ur'.*félév\s+(\d+)\.\s+hete.*', text)
|
||||
if regex:
|
||||
week = regex.groups()[0]
|
||||
else:
|
||||
regex = re.match(ur'.*nyár\s+(\d+)\.\s+hete.*', text)
|
||||
if regex:
|
||||
raise EnvironmentError('Nyar van!')
|
||||
else:
|
||||
raise RuntimeError('Baj van a KTH oldalan!')
|
||||
|
||||
return int(week) - 1
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
print(get_week())
|
Loading…
Reference in New Issue
Block a user