Feature Request: Get object's raw source

[lpinilla]

Hi,

How are you? Thanks of all, thanks for doing the port of the tool, I was thinking of doing it myself and I found someone else did it :smile:

I was wondering if there was a way to get the raw source of an objects, let's say a contract or function.

I was thinking in something like:

sourceUnitObject = parser.objectify(sourceUnit)
raw_text = sourceUnitObject.contracts['MyContract'].raw
print(raw_text)
>>'contract MyContract { ..'

Thanks!

[seyyedaliayati]

This is indeed a very useful feature. I started working on it.

[seyyedaliayati]

This is indeed a very useful feature. I started working on it.

Here is my solution:

1. Set loc=True in parse function.

2. Use the following helper function:

   def get_content_between_positions(file_path, positions):
   start_line = positions['start']['line']
   start_column = positions['start']['column']
   end_line = positions['end']['line']
   end_column = positions['end']['column']
   
   with open(file_path, 'r') as file:
       lines = file.readlines()
   
   content = ''
   for line_number, line in enumerate(lines, 1):
       if start_line <= line_number <= end_line:
           if line_number == start_line:
               content += line[start_column:]
           elif line_number == end_line:
               content += line[:end_column+1]
           else:
               content += line
   
   return content

3. use ._node.loc. to access the location dictionary.

   
   from solidity_parser import parser, objectify, visit
   source_unit = parser.parse_file(sample_test_file, loc=True)
   source_unit_obj = objectify(source_unit)

contracts = source_unit_obj.contracts

for contract in contracts.values(): print(get_content_between_positions(sample_test_file, contract.functions['']._node.loc))

The output will be the context of the function! Enjoy it :)