step1.py

@@ -1,35 +1,30 @@
-# add ...
 # -*- coding: utf-8 -*-
-# author: youxinyu
-# githb: yogayu
 
 # added by daiao
 
 from pyquery import PyQuery as pq
 
-f = open("step2/index.html", encoding='utf-8')
+f = open("step1/index.html", encoding='utf-8')
+# f = open("../step1/result.html", encoding='utf-8')
+
 html = f.read()
 f.close()
 
 doc = pq(html)
+
 tag = input()
-result = ""
-if tag == "meta":
-    tag = doc('meta:nth-child(4)')
-    attr = input()
-    result = tag.attr(attr)
-else:
-    tag = doc(tag)
-    result = doc(tag).text()
-print (result, end='')
 
+result = ''
+if tag == 'h1' or 'p':
+    result = doc(tag).text()
 
 if tag == 'align':
     h1Align= doc('h1').attr('align')
     pAlign= doc('p').attr('align')
     result = (h1Align and pAlign)
 
+if tag == 'bgcolor':
+    result = doc('body').attr('bgcolor')
 
 # print(result),
 print (result, end='')
-