How can I write a multiline code with %stata in python?

[John Williamss]

Suppose I write the following code in python:

%loading pystata as usual
import stata_setup
stata_setup.config("C:/Program Files/Stata17", "mp")
from pystata import stata
from sfi import Macro

for ii in [1,2]:
 Macro.setLocal('kk', str(ii))
 %stata forvalues jj=1/`kk' {
 %stata disp `jj'
 %stata }

But this obviously does not work as the %stata command is designed to be a one-line. What is the workaround here?
(The example is just intended to be an MWE. The actual code (in python) is more than a thousand lines and should be able to run with one click and no more interaction.)

[FernandoRios]

You need magic
%%stata
multiple Stata code lines

[FernandoRios]

You need magic
%%stata
multiple Stata code lines

[John Williamss]

You need magic
%%stata
multiple Stata code lines

No. Magic %% does not work with other python commands I've tried that already. For example, this does not work:

 %loading pystata as usual
import stata_setup
stata_setup.config("C:/Program Files/Stata17", "mp")
from pystata import stata
from sfi import Macro  

for ii in [1,2]:  
 Macro.setLocal('kk', str(ii))
 %%stata  
 forvalues jj=1/`kk'
 {  
 disp `jj'
 }

[Zhao Xu (StataCorp)]

No. Magic %% does not work with other python commands I've tried that already. For example, this does not work:

%loading pystata as usual
import stata_setup
stata_setup.config("C:/Program Files/Stata17", "mp")
from pystata import stata
from sfi import Macro

for ii in [1,2]:
Macro.setLocal('kk', str(ii))
%%stata
forvalues jj=1/`kk'
{
disp `jj'
}

Hi John Williams,

You are right. Cell magic %% will not run within a Python loop. Alternatively, you can use the run() function within the stata module of pystata to run a block of code as follows:

from pystata import stata
from sfi import Macro

Then run

for ii in [1,2]:
    Macro.setLocal('kk', str(ii))
    stata.run('''
        forvalues jj=1/`kk' {
            disp `jj'
        }
    ''')

The output is:

. 
.         forvalues jj=1/`kk' {
  2.             disp `jj'
  3.         }
1

.     
. 

. 
.         forvalues jj=1/`kk' {
  2.             disp `jj'
  3.         }
1
2

In this way, you can mix a block of Python statements and a block of Stata code together.

[John Williamss]

Hi John Williams,

You are right. Cell magic %% will not run within a Python loop. Alternatively, you can use the run() function within the stata module of pystata to run a block of code as follows:

from pystata import stata
from sfi import Macro

Then run

for ii in [1,2]:
Macro.setLocal('kk', str(ii))
stata.run('''
forvalues jj=1/`kk' {
disp `jj'
}
''')

The output is:

.
. forvalues jj=1/`kk' {
2. disp `jj'
3. }
1

.
.

.
. forvalues jj=1/`kk' {
2. disp `jj'
3. }
1
2

In this way, you can mix a block of Python statements and a block of Stata code together.

Thanks!