css "^" as "." xpath symbol to use css "^ >" to get immediate children

Author: sortafreel

.gitignore

@@ -5,4 +5,6 @@
 /dist
 /docs/_build
 /.coverage
-.idea
+.idea
+/venv
+*.vscode

cssselect/parser.py

@@ -400,8 +400,8 @@ def parse_simple_selector(stream, inside_negation=False):
     stream.skip_whitespace()
     selector_start = len(stream.used)
     peek = stream.peek()
-    if peek.type == 'IDENT' or peek == ('DELIM', '*'):
-        if peek.type == 'IDENT':
+    if peek.type == 'IDENT' or peek == ('DELIM', '*') or peek == ('DELIM', '^'):
+        if peek.type == 'IDENT' or peek == ('DELIM', '^'):
             namespace = stream.next().value
         else:
             stream.next()

cssselect/xpath.py

@@ -187,6 +187,14 @@ def css_to_xpath(self, css, prefix='descendant-or-self::'):
             The equivalent XPath 1.0 expression as an Unicode string.
 
         """
+        # no prefix if css immediate children (example: css "^ > div" to xpath "./div")
+        child_re = r'^[ \t\r\n\f]*\^[ \t\r\n\f]*>'
+        if re.match(child_re, css):
+            prefix = ''
+            # prefix = 'child::'
+            # css = re.sub(child_re, '', css)
+            # print('*' * 50)
+            # print(css)
         return ' | '.join(self.selector_to_xpath(selector, prefix,
                                                  translate_pseudo_elements=True)
                           for selector in parse(css))
@@ -332,6 +340,9 @@ def xpath_element(self, selector):
         if not element:
             element = '*'
             safe = True
+        if element == '^':
+            element = '.'
+            safe = True
         else:
             safe = is_safe_name(element)
             if self.lower_case_element_names: