opsdroid-taginfo/__init__.py at main · mahdi1234/opsdroid-taginfo · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
import random, urllib.request, json, sys, re

from opsdroid.matchers import match_regex
from opsdroid.skill import Skill

usage = ("<pre>Usage: '!tg key=value' or 'value'\ne.g. '!tg highway=steps' or '!tg highway=*' or 'steps'</pre>")

class TagInfo(Skill):
	# match the user input, it can be anywhere, without .* it would have to be at the start of the message which wouldn't work for my matrix-xmpp bridge
	@match_regex(r'.*?!tg (?P<input>.*)', case_sensitive=False)
	async def hello(self, message):
		# get user input
		input_text = message.regex.group('input')
		try:

			# invoke help on help keyword
			if re.search(r'^help$/i', input_text):
				await message.respond(usage)

			# catch spaces
			elif re.search(r' ', input_text):
				await message.respond(usage)

			# querying wildcard e.g. highway=*
			elif re.search(r'^[a-zA-Z:_]+=\*$', input_text):
				osm_key,osm_value=input_text.split('=')
				text = f"<pre>\n### Occurence of {input_text} ###\n\n"
				# get results via taginfo API
				with urllib.request.urlopen("https://taginfo.openstreetmap.cz/api/4/key/values?key=" + osm_key + "&filter=all&lang=cz&sortname=count&sortorder=desc&page=1&rp=10") as url:
					data = json.loads(url.read().decode())
					for element in data['data']:
						# taginfo API returns empty string instead of * on website, so I had to convert it back to * and str.replace() doesn't behave as I expected thus used .re
						text += f"{element['value']} - {str(element['count'])}\n"
					# return everything at once as answer to chat
					text += "</pre>"
					await message.respond(text)

			# querying exact combination e.g. highway=steps
			elif re.search(r'^[a-zA-Z:_]+=[a-zA-Z:_]+$', input_text):
				osm_key,osm_value=input_text.split('=')
				text = f"<pre>\n### Occurence of {input_text} ###\n\n"
				# get results via taginfo API
				with urllib.request.urlopen("https://taginfo.openstreetmap.cz/api/4/tag/stats?key=" + osm_key + "&value=" + osm_value) as url:
					data = json.loads(url.read().decode())
					# and feed it to text string, otherwise there's problem with mixing types
					for element in data['data']:
						text += f"{element['type']} - {str(element['count'])}\n"

				text += "\n### and most common tag combinations ###\n\n"
				# here we go with another taginofo query
				with urllib.request.urlopen("https://taginfo.openstreetmap.cz/api/4/tag/combinations?key=" + osm_key + "&value=" + osm_value + "&sortorder=desc") as url:
					data = json.loads(url.read().decode())
					for element in data['data']:
						# taginfo API returns empty string instead of * on website, so I had to convert it back to * and str.replace() doesn't behave as I expected thus used .re
						text += f"{element['other_key']}:{re.sub('^$', ' *', element['other_value'])} - {str(element['together_count'])}\n"
					# return everything at once as answer to chat
					text += "</pre>"
					await message.respond(text)

			# querying value only
			elif re.search(r'^[a-zA-Z:_]+$', input_text):
				text = f"<pre>\n### Occurence of value {input_text} ###\n\n"
				# get results via taginfo API
				with urllib.request.urlopen("https://taginfo.openstreetmap.cz/api/4/search/by_value?query=" + input_text + "&sortname=count_all&sortorder=desc&page=1&rp=10&format=json_pretty") as url:
					data = json.loads(url.read().decode())
					for element in data['data']:
						# taginfo API returns empty string instead of * on website, so I had to convert it back to * and str.replace() doesn't behave as I expected thus used .re
						text += f"{element['key']} = {element['value']} - {str(element['count_all'])}\n"
					# return everything at once as answer to chat
					text += "</pre>"
					await message.respond(text)

			else:
				await message.respond(usage)

		except:
			# if parsing of user params fails give usage info
				await message.respond(usage)